lagy commited on
Commit
b39018a
·
verified ·
1 Parent(s): f2bcf60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -53,12 +53,12 @@ def generate(model,tokenizer,instruction,context,streamer):
53
  @st.cache_resource
54
  def get_model():
55
  # Create a model of the specified type.
56
- return AutoModelForCausalLM.from_pretrained("lagy/carballo_finetuned")
57
 
58
  @st.cache_resource
59
  def get_tokenizer():
60
  # Create a model of the specified type.
61
- return AutoTokenizer.from_pretrained("lagy/carballo_finetuned")
62
 
63
  st.write(" ")
64
 
 
53
  @st.cache_resource
54
  def get_model():
55
  # Create a model of the specified type.
56
+ return AutoModelForCausalLM.from_pretrained("lagy/carballo-1.3b-finetuned")
57
 
58
  @st.cache_resource
59
  def get_tokenizer():
60
  # Create a model of the specified type.
61
+ return AutoTokenizer.from_pretrained("lagy/carballo-1.3b-finetuned")
62
 
63
  st.write(" ")
64