Update app.py

app.py

@@ -14,7 +14,7 @@ def predict(input, history=[]):
     new_user_input_ids = tokenizer.encode(input + tokenizer.eos_token, return_tensors="pt")
 
     # Append the new user input tokens to the chat history
-    bot_input_ids = torch.cat([torch.tensor(history), new_user_input_ids], dim=-1) if history else new_user_input_ids
+    bot_input_ids = torch.cat([torch.tensor(history, dtype=torch.long), new_user_input_ids], dim=-1) if history else new_user_input_ids
 
     # Generate a response
     chat_history_ids = model.generate(bot_input_ids, max_length=200, pad_token_id=tokenizer.eos_token_id)
@@ -23,7 +23,7 @@ def predict(input, history=[]):
     response = tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True)
 
     # Update the history with the new tokens
-    new_history = chat_history_ids.tolist()
+    new_history = chat_history_ids[0].tolist()
 
     return response, new_history
 
@@ -32,7 +32,7 @@ gr.Interface(
     title=title,
     description=description,
     examples=examples,
-    inputs=[gr.components.Textbox(lines=2, placeholder="Enter your message here..."), "state"],
-    outputs=[gr.components.Textbox(), "state"],
+    inputs=[gr.inputs.Textbox(lines=2, placeholder="Enter your message here..."), gr.inputs.State()],
+    outputs=[gr.outputs.Textbox(), gr.outputs.State()],
     theme="finlaymacklon/boxy_violet",
 ).launch()