Update app.py

app.py

@@ -60,10 +60,10 @@ def get_response(prompt, model_name, user_type):
 
     # Define different prompt templates based on user type
     user_type_templates = {
-        "Professional Chef": f"As a professional chef, {prompt}\nAnswer:",
-        "Home Cook": f"As a home cook, {prompt}\nAnswer:",
+        "Professional": f"As a professional chef, {prompt}\nAnswer:",
         "Beginner": f"Explain in simple terms: {prompt}\nAnswer:",
-        "Food Enthusiast": f"As a food enthusiast, {prompt}\nAnswer:"
+        "Intermediate": f"As an intermediate cook, {prompt}\nAnswer:",
+        "Expert": f"As an expert chef, {prompt}\nAnswer:"
     }
 
     # Get the appropriate prompt based on user type
@@ -74,11 +74,11 @@ def get_response(prompt, model_name, user_type):
         return_tensors='pt',
         padding=True,
         truncation=True,
-        max_length=300  # Increased length for larger inputs
+        max_length=512  # Increased max length for larger inputs
     ).to(device)
 
-    # Reduce max_new_tokens for faster response time
-    max_new_tokens = 50  # Reduced to speed up response time
+    # Increase max_new_tokens for longer responses
+    max_new_tokens = 200  # Increase this to allow more content in response
 
     with torch.no_grad():
         output = model.generate(
@@ -193,10 +193,10 @@ body {
         response = gr.Textbox(
             label="Response",
             placeholder="Your answer will appear here...",
-            lines=10,
+            lines=15,  # Increased lines for a longer response display
             interactive=False,
             show_copy_button=True,
-            max_lines=12
+            max_lines=20  # Allow for more lines if the response is lengthy
         )
 
     submit_button.click(fn=process_input, inputs=[prompt, model_name, user_type], outputs=response)