Update app.py

app.py

@@ -8,8 +8,11 @@ description = "A State-of-the-Art Large-scale Pretrained Response generation mod
 examples = [["How are you?"]]
 
 
-tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
-model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
+# tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
+# model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
+
+tokenizer = AutoTokenizer.from_pretrained("Salesforce/xgen-7b-8k-base", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("Salesforce/xgen-7b-8k-base", torch_dtype=torch.bfloat16)
 
 
 def predict(input, history=[]):