Update app.py

app.py

@@ -47,9 +47,9 @@ def generate_text(tokenizer, model, device, prompt, max_length=100,
     )
 
     # Convert the generated text back to a string
-    generated_text = tokenizer.decode(ids, skip_special_tokens=True) for ids in output
+    generated_text = [tokenizer.decode(ids, skip_special_tokens=True) for ids in output]
 
-    return generated_text
+    return generated_text[0]
 
 def gradio_generate_text(prompt, max_length=100, num_return_sequences=1, top_p=0.95, temperature=0.7):
     generated_text = generate_text(tokenizer, model, device, prompt, max_length, num_return_sequences, top_p, temperature)