Update app.py

app.py

@@ -1,7 +1,5 @@
-import gradio as gr
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
-# Load the pre-trained DialoGPT model and tokenizer
 tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
 model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
 
@@ -16,11 +14,10 @@ def chatbot(input_text):
     response_text = tokenizer.decode(response_ids[:, input_ids.shape[-1]:][0], skip_special_tokens=True)
     return response_text
 
-# Create a Gradio interface for the chatbot
-gr.Interface(
-    fn=chatbot,
-    inputs="text",
-    outputs="text",
-    title="AI Chatbot",
-    description="Ask me anything!"
-).launch()
+while True:
+    user_input = input("You: ")
+    if user_input.lower() == 'exit':
+        print("Chatbot: Goodbye!")
+        break
+    response = chatbot(user_input)
+    print("Chatbot:", response)