Update app.py

app.py

@@ -1,23 +1,23 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 
-model_name = "mistralai/Mistral-7B-Instruct-v0.1"
+model_name = "tiiuae/falcon-rw-1b"
 
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
+model = AutoModelForCausalLM.from_pretrained(model_name)
 
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
 def chat(user_input, history=[]):
-    prompt = f"[INST] {user_input} [/INST]"
-    result = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.7)
+    prompt = f"{user_input}"
+    result = pipe(prompt, max_new_tokens=200, do_sample=True, temperature=0.8)
     response = result[0]["generated_text"].replace(prompt, "").strip()
     history.append((user_input, response))
     return history, history
 
 gr.ChatInterface(
     fn=chat,
-    title="🤖 Mobe 2.0 - Personal AI by Ronobir",
-    description="Powered by Mistral 7B | 100% Free",
+    title="🤖 Mobe 2.0 – Public AI (No login needed)",
+    description="Now using Falcon 1B - 100% free",
     theme="soft"
 ).launch()