6Genix commited on
Commit
4e24e07
·
1 Parent(s): cf2cc49

Zephyr had incorrect folder

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -47,8 +47,8 @@ def load_model_engineer():
47
  @st.cache_resource
48
  def load_model_analyst():
49
  # Analyst: zephyr-7b-beta
50
- tokenizerA = AutoTokenizer.from_pretrained("zephyr-ai/zephyr-7b-beta")
51
- modelA = AutoModelForCausalLM.from_pretrained("zephyr-ai/zephyr-7b-beta")
52
  return tokenizerA, modelA
53
 
54
  # Load models
 
47
  @st.cache_resource
48
  def load_model_analyst():
49
  # Analyst: zephyr-7b-beta
50
+ tokenizerA = AutoTokenizer.from_pretrained("HuggingFaceH4/zephyr-7b-beta")
51
+ modelA = AutoModelForCausalLM.from_pretrained("HuggingFaceH4/zephyr-7b-beta")
52
  return tokenizerA, modelA
53
 
54
  # Load models