Abbeite commited on
Commit
c3834df
1 Parent(s): 6a77361

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ logging.basicConfig(level=logging.CRITICAL)
9
  # Cache the model and tokenizer to avoid reloading it every time
10
  @st.experimental_singleton
11
  def load_model():
12
- model_name = "Abbeite/trail_wl" # Replace with your actual model name
13
  tokenizer = AutoTokenizer.from_pretrained(model_name)
14
  model = AutoModelForCausalLM.from_pretrained(model_name)
15
  return model, tokenizer
 
9
  # Cache the model and tokenizer to avoid reloading it every time
10
  @st.experimental_singleton
11
  def load_model():
12
+ model_name = "Abbeite/chest_and_physical_limitations2" # Replace with your actual model name
13
  tokenizer = AutoTokenizer.from_pretrained(model_name)
14
  model = AutoModelForCausalLM.from_pretrained(model_name)
15
  return model, tokenizer