Bisher commited on
Commit
695aa89
·
verified ·
1 Parent(s): 076acc6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ pipe = pipeline("text2text-generation", model="basharalrfooh/Fine-Tashkeel")
7
 
8
  def diacratize_text(text):
9
  generated_text = pipe(text)
10
- return' '.join([s['generated_text'] for s in generated_text])
11
 
12
  demo = gr.Interface(
13
  fn=diacratize_text,
 
7
 
8
  def diacratize_text(text):
9
  generated_text = pipe(text)
10
+ return pipe(text)[0]['generated_text']
11
 
12
  demo = gr.Interface(
13
  fn=diacratize_text,