Update app.py

app.py

@@ -18,7 +18,7 @@ def generate_qa(context, temperature, top_p):
         top_p=top_p
     )
     
-    generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
+    generated_text = tokenizer.batch_decode(output, skip_special_tokens=True)
     return generated_text
 
 # Create the Gradio interface with sliders for temperature and top-p