Spaces:
Running
Running
fixed bug with check_model
Browse files
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 |
-
|
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:
|