Update app.py

app.py

@@ -1,11 +1,11 @@
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
-#set up the model (large version of DialoGPT)
-tokenizer = AutoTokenizer.from_pretrained("timdettmers/guanaco-65b-merged")
-model = AutoModelForCausalLM.from_pretrained("timdettmers/guanaco-65b-merged")
 
-#Defining a predict function
+#set up the model
+tokenizer = AutoTokenizer.from_pretrained("OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5")
+model = AutoModelForCausalLM.from_pretrained("OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5")
 
+#Defining a predict function
 def predict(input, history=[]):
     # tokenize the new input sentence
     new_user_input_ids = tokenizer.encode(input + tokenizer.eos_token, return_tensors='pt')
@@ -22,11 +22,10 @@ def predict(input, history=[]):
     return response, history
 
 #creating a gradio interface
-
 import gradio as gr
 
 demo = gr.Interface(fn=predict,
-             examples=["How many"],
+             examples=["How many birds exist on Earth"],
              inputs=["text", "state"],
              outputs=["chatbot", "state"])