Update app.py

app.py

@@ -2,8 +2,8 @@ import gradio as gr
 from transformers import pipeline
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-tokenizer = AutoTokenizer.from_pretrained("facebook/galactica-1.3b")
-model = AutoModelForCausalLM.from_pretrained("facebook/galactica-1.3b")
+tokenizer = AutoTokenizer.from_pretrained("facebook/galactica-125m")
+model = AutoModelForCausalLM.from_pretrained("facebook/galactica-125m")
 text2text_generator = pipeline("text-generation", model=model, tokenizer=tokenizer, num_workers=2)
 
 def predict(text, max_length=64, temperature=0.7, do_sample=True):