Update app.py

app.py

@@ -7,16 +7,17 @@ generator = pipeline("text-generation", model=model_name)
 
 # Inference function
 def generate_response(prompt):
-    # Generate text with specific parameters
+    # Adjust the prompt to guide the model
+    modified_prompt = f"Answer the question directly: {prompt}"
     response = generator(
-        prompt,
-        max_length=150,  # Increase max length for more comprehensive responses
+        modified_prompt,
+        max_length=150,  # Maintain this to allow for longer responses
         num_return_sequences=1,
-        temperature=0.7,  # Lower for more deterministic responses
-        top_k=50,         # Consider the top 50 tokens for diversity
-        top_p=0.95        # Cumulative probability for diversity
+        temperature=0.7,
+        top_k=50,
+        top_p=0.95
     )
-    return response[0]['generated_text'].strip()  # Clean up the output
+    return response[0]['generated_text'].strip()
 
 # Gradio interface
 interface = gr.Interface(