Update app.py

app.py

@@ -7,11 +7,18 @@ def generate_text(input_text, history):
     # Initialize the pipeline with specified parameters
     pipe = pipeline("text-generation", model="ngxson/MiniThinky-v2-1B-Llama-3.2", max_new_tokens=1024, temperature=0.7, do_sample=True)
     
+    # Prepare the messages
+    messages = [{"role": "user", "content": input_text}]
+    
     # Generate response
-    response = pipe(input_text, max_new_tokens=1024, temperature=0.7, do_sample=True)
+    response = pipe(messages)
+    
+    # Extract the generated text
+    response_text = response[0]["generated_text"]
     
     # Append user and assistant messages to history
-    history.append((input_text, response[0]['generated_text']))
+    history.append({"role": "user", "content": input_text})
+    history.append({"role": "assistant", "content": response_text})
     
     # Return updated history
     return history
@@ -24,7 +31,7 @@ demo = gr.Interface(
         gr.JSON(value=[], visible=False)
     ],
     outputs=[
-        gr.Chatbot(label="Chat History")
+        gr.Chatbot(type='messages')
     ],
     title="MiniThinky Text Generator"
 )