Update app.py

app.py

@@ -8,7 +8,7 @@ pipe = pipeline('text-generation', model=model,tokenizer = tokenizer)
 
 @spaces.GPU(duration=120)
 def generate(prompt):
-    return pipe(prompt)['generated_text'][0]
+    return pipe(prompt)[0]['generated_text']
 
 gr.Interface(
     fn=generate,