Update app.py

app.py

@@ -2,8 +2,8 @@ import torch
 import gradio as gr
 from transformers import AutoTokenizer, OPTForCausalLM
 
-tokenizer = AutoTokenizer.from_pretrained("facebook/galactica-6.7b")
-model = OPTForCausalLM.from_pretrained("facebook/galactica-6.7b",cache_dir='${PWD}/cache',offload_folder='${PWD}/offload/')
+tokenizer = AutoTokenizer.from_pretrained("facebook/galactica-125m")
+model = OPTForCausalLM.from_pretrained("facebook/galactica-125m",cache_dir='${PWD}/cache',offload_folder='${PWD}/offload/')
 
 def process_text(text):
     input_ids = tokenizer(text, return_tensors="pt").input_ids