Update app.py

app.py

@@ -1,19 +1,17 @@
-import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
-import torch
-import json
 
 tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
 model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
 
-def predict(input_text, state=[]):
+
+def predict(input, history=[]):
     # tokenize the new input sentence
     new_user_input_ids = tokenizer.encode(
-        input_text + tokenizer.eos_token, return_tensors="pt"
+        input + tokenizer.eos_token, return_tensors="pt"
     )
 
     # append the new user input tokens to the chat history
-    bot_input_ids = torch.cat([torch.LongTensor(state), new_user_input_ids], dim=-1)
+    bot_input_ids = torch.cat([torch.LongTensor(history), new_user_input_ids], dim=-1)
 
     # generate a response
     history = model.generate(
@@ -21,13 +19,15 @@ def predict(input_text, state=[]):
     ).tolist()
 
     # convert the tokens to text, and then split the responses into lines
-    response = tokenizer.decode(history[0]).split(" ")
+    response = tokenizer.decode(history[0]).split("<|endoftext|>")
+    # print('decoded_response-->>'+str(response))
     response = [
         (response[i], response[i + 1]) for i in range(0, len(response) - 1, 2)
     ]  # convert to tuples of list
+    # print('response-->>'+str(response))
     return response, history
 
-def read_json_file(file_path):
+def read_json_file(file_path): #read json file test
     with open(file_path, 'r') as file:
         data = json.load(file)
     return data
@@ -46,15 +46,14 @@ def main():
 if __name__ == "__main__":
     main()
 
-title = "Chat with AI"
-description = "This AI chatbot can respond to your queries using a DialoGPT-based model and additional learned responses."
-examples = [["Hello, how are you?"], ["What is the weather today?"]]
+
 
 gr.Interface(
     fn=predict,
     title=title,
     description=description,
-    inputs=["text", "text"],
-    outputs=["text", "text"],
+    examples=examples,
+    inputs=["text", "state"],
+    outputs=["chatbot", "state"],
     theme="finlaymacklon/boxy_violet",
-).launch()
+).launch()