Update app.py

app.py

@@ -78,19 +78,21 @@
 # # ).launch()
 
 
-
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
 import torch
 
 
-title = "🤖AI ChatBot"
-description = "A State-of-the-Art Large-scale Pretrained Response generation model (DialoGPT)"
+title = "🦅Falcon 🗨️ChatBot"
+description = "Falcon-RW-1B is a 1B parameters causal decoder-only model built by TII and trained on 350B tokens of RefinedWeb."
 examples = [["How are you?"]]
 
 
 tokenizer = AutoTokenizer.from_pretrained("Cogwisechat/falcon-7b-finance")
-model = AutoModelForCausalLM.from_pretrained("Cogwisechat/falcon-7b-finance")
+model = AutoModelForCausalLM.from_pretrained(
+    "Cogwisechat/falcon-7b-finance",
+    trust_remote_code=True,
+)
 
 
 def predict(input, history=[]):
@@ -125,4 +127,4 @@ gr.Interface(
     inputs=["text", "state"],
     outputs=["chatbot", "state"],
     theme="finlaymacklon/boxy_violet",
-).launch()
+).launch()