duatanzeel commited on
Commit
3ad1da4
·
verified ·
1 Parent(s): 86d2e8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -14,12 +14,13 @@ def load_model():
14
  try:
15
  model = pipeline(
16
  "text-generation",
17
- model="mistralai/Mixtral-8x7B-Instruct-v0.1",
18
  token=HF_TOKEN
19
  )
20
  return model
21
  except Exception as e:
22
- st.error(" Failed to load Mixtral model. Falling back to GPT-2.")
 
23
  return pipeline("text-generation", model="gpt2")
24
 
25
  model = load_model()
 
14
  try:
15
  model = pipeline(
16
  "text-generation",
17
+ model="tiiuae/falcon-7b-instruct", # lightweight, non-gated
18
  token=HF_TOKEN
19
  )
20
  return model
21
  except Exception as e:
22
+ print(f"Model load failed: {e}")
23
+ st.error("❌ Failed to load model.")
24
  return pipeline("text-generation", model="gpt2")
25
 
26
  model = load_model()