Abbeite commited on
Commit
3c82543
1 Parent(s): c3834df

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/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
 
9
  # Cache the model and tokenizer to avoid reloading it every time
10
  @st.experimental_singleton
11
  def load_model():
12
+ model_name = "NousResearch/Llama-2-7b-chat-hf" # Replace with your actual model name
13
  tokenizer = AutoTokenizer.from_pretrained(model_name)
14
  model = AutoModelForCausalLM.from_pretrained(model_name)
15
  return model, tokenizer