Update app.py

app.py

@@ -59,8 +59,10 @@ class MissionGenerator:
         outputs = self.model.generate(
             inputs["input_ids"],
             max_length=256,
-            num_beams=4,
-            temperature=0.7,
+            do_sample=True,  # Enable sampling
+            temperature=0.7,  # Control randomness
+            top_p=0.9,       # Nucleus sampling
+            num_beams=4,     # Beam search
             no_repeat_ngram_size=2
         )
         
@@ -92,7 +94,8 @@ def create_gradio_interface():
     
     def process_input(user_input: str, history: List[Dict]) -> tuple[List[Dict], str]:
         chat_response, formatted_output = generator.generate_response(user_input)
-        history.append({"user": user_input, "bot": chat_response})
+        history.append({"role": "user", "content": user_input})
+        history.append({"role": "assistant", "content": chat_response})
         return history, formatted_output
     
     with gr.Blocks() as interface: