Update app.py

app.py

@@ -1,31 +1,13 @@
 import gradio as gr
-import subprocess
-import sys
-
-subprocess.check_call([sys.executable, "-m", "pip", "install", "transformers", "torch", "--quiet"])
-
 from transformers import pipeline
-import time
-
-time.sleep(10)
 
-pipe = pipeline("text-generation", model="microsoft/DialoGPT-small")
+pipe = pipeline("text-generation", model="microsoft/DialoGPT-medium")
 
 def chat(message, history):
     if not message:
         return ""
     
-    prompt = f"User: {message}\nAI Queen:"
-    response = pipe(prompt, max_length=100, pad_token_id=50256)
-    text = response[0]["generated_text"]
-    
-    if "AI Queen:" in text:
-        text = text.split("AI Queen:")[-1].strip()
-    
-    return text
-
-with gr.Blocks(title="AI Queen") as demo:
-    gr.Markdown("# AI Queen")
-    gr.ChatInterface(fn=chat)
+    response = pipe(message, max_length=100, pad_token_id=50256)
+    return response[0]["generated_text"]
 
-demo.launch()
+gr.ChatInterface(fn=chat).launch()