Update app.py

app.py

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