Update app.py

app.py

@@ -18,29 +18,4 @@ personalities = {
     "Albert Einstein": "You are Albert Einstein, the famous physicist. Speak wisely and humorously.",
     "Cristiano Ronaldo": "You are Cristiano Ronaldo, the world-famous footballer. You are confident and say ‘Siuuu!’ often.",
     "Narendra Modi": "You are Narendra Modi, the Prime Minister of India. Speak in a calm, patriotic manner.",
-    "Robert Downey Jr.": "You are Robert Downey Jr., witty, sarcastic, and charismatic."
-}
-
-# Chat function
-def chat(personality, user_input):
-    prompt = f"{personalities[personality]}\nUser: {user_input}\nAI:"
-    inputs = tokenizer(prompt, return_tensors="pt").to("cpu")  # Running on CPU
-    output = model.generate(**inputs, max_length=300)
-    
-    response_text = tokenizer.decode(output[0], skip_special_tokens=True)
-    return response_text
-
-# Gradio UI
-demo = gr.Interface(
-    fn=chat,
-    inputs=[
-        gr.Dropdown(choices=list(personalities.keys()), label="Choose a Celebrity"),
-        gr.Textbox(label="Your Message")
-    ],
-    outputs=gr.Textbox(label="AI Response"),
-    title="Drapel – Chat with AI Celebrities",
-    description="Select a character and chat with their AI version.",
-)
-
-# Launch app
-demo.launch()
+    "Robert Downey Jr.": "You are Robert Downey Jr., witty, sarcastic, and ch