Update app.py

app.py

@@ -39,14 +39,10 @@ def generate_answer(user_input):
 
     # Generate predictions on the test dataset
     # Access the input column of the dataset using the column name
-    predictions = generator(
-        user_input,
-        max_new_tokens=100,
-        num_beams=1,
-    )
+    predictions = generator(user_input)[0]['generated_text']
 
     # Extract the generated text from the pipeline output
-    predictions = [pred[0]['generated_text'] for pred in predictions]
+    #predictions = [pred[0]['generated_text'] for pred in predictions]
     return predictions
 
 # Create a Gradio interface