JPBianchi commited on
Commit
55a0f00
1 Parent(s): 2e4b5f4

fixed bug with check_model

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -222,7 +222,6 @@ def check_model(model_name_or_path):
222
  # let's download the model, HF is not limited in space like Streamlit.io
223
  download_model(model_name_or_path, model_local_path)
224
 
225
- return model_local_path
226
 
227
  #%% instantiate Weaviate client
228
  def get_weaviate_client(api_key, url, model_name_or_path, openai_api_key):
@@ -327,7 +326,7 @@ def main():
327
  else:
328
  st.write("Finetuning not available on Streamlit online because of space limitations")
329
 
330
- model_name_or_path = check_model(model_name_or_path)
331
  try:
332
  client, available_classes = get_weaviate_client(Wapi_key, url, model_name_or_path, openai_api_key)
333
  except Exception as e:
 
222
  # let's download the model, HF is not limited in space like Streamlit.io
223
  download_model(model_name_or_path, model_local_path)
224
 
 
225
 
226
  #%% instantiate Weaviate client
227
  def get_weaviate_client(api_key, url, model_name_or_path, openai_api_key):
 
326
  else:
327
  st.write("Finetuning not available on Streamlit online because of space limitations")
328
 
329
+ check_model(model_name_or_path)
330
  try:
331
  client, available_classes = get_weaviate_client(Wapi_key, url, model_name_or_path, openai_api_key)
332
  except Exception as e: