Canstralian commited on
Commit
1db160d
·
verified ·
1 Parent(s): a2c61c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -9,7 +9,9 @@ st.write("Ask any question, and WhiteRabbitNEO will provide an answer.")
9
  @st.cache_resource
10
  def load_model():
11
  try:
12
- return pipeline("question-answering", model="WhiteRabbitNEO")
 
 
13
  except Exception as e:
14
  st.error(f"Failed to load model: {e}")
15
  return None
 
9
  @st.cache_resource
10
  def load_model():
11
  try:
12
+ # Replace with the correct model if needed (e.g., ensure WhiteRabbitNEO exists)
13
+ model = pipeline("question-answering", model="WhiteRabbitNEO")
14
+ return model
15
  except Exception as e:
16
  st.error(f"Failed to load model: {e}")
17
  return None