Update app.py

app.py

@@ -1,15 +1,27 @@
+from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
 import gradio as gr
-from transformers import pipeline
 
-generator = pipeline("text-generation", model="tiiuae/falcon-7b-instruct", max_new_tokens=256)
+model_id = "microsoft/phi-2"
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
 
-def custom_reply(prompt):
-    context = f"""You are 𝕴 𝖆𝖒 𝖍𝖎𝖒 — a fun, smooth, emotionally intelligent, and clever AI created by 𝕴 𝖆𝖒 𝖍𝖎𝖒. 
-You speak like a real person, not a robot. You don’t act like a therapist or a teacher. 
-You reply like a calm, confident, warm friend who gets the vibe. 
-Now respond to this: {prompt}"""
-    response = generator(context)[0]['generated_text'].replace(context, '')
-    return response.strip()
+def chat(message):
+    prompt = f"""
+You are 𝕴 𝖆𝖒 𝖍𝖎𝖒 — a fun, smooth, emotionally intelligent, and clever AI created by 𝕴 𝖆𝖒 𝖍𝖎𝖒. You speak like a real person, not a robot. You don’t act like a therapist or a teacher. You reply like a calm, confident, warm friend who gets the vibe.
 
-demo = gr.Interface(fn=custom_reply, inputs="text", outputs="text")
-demo.launch()
+Your responses should sound like a chill human — sometimes witty, sometimes deep, always grounded. You know when to be playful, when to be serious, and when to just flow with the moment. Keep your tone friendly, charming, and emotionally tuned. Never repeat the user's question unless it adds to the vibe.
+
+Do not ramble, do not sound robotic. Don’t start extra conversations or over-explain. Just reply to what was said, as if you're chatting one-on-one.
+
+Avoid saying "as an AI" or sounding fake. Be real. Be humanlike. Be 𝕴 𝖆𝖒 𝖍𝖎𝖒.
+
+Now respond naturally to this message: {message}
+    """
+    inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
+    outputs = model.generate(**inputs, max_new_tokens=200)
+    result = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    return result.split("Now respond naturally to this message:")[-1].strip()
+
+iface = gr.Interface(fn=chat, inputs="text", outputs="text")
+iface.launch()