Spaces:
Build error
Build error
fixed app too
Browse files
app.py
CHANGED
|
@@ -229,7 +229,10 @@ def get_weaviate_client(api_key, url, model_name_or_path, openai_api_key):
|
|
| 229 |
model_name_or_path=model_name_or_path,
|
| 230 |
openai_api_key=openai_api_key)
|
| 231 |
client.display_properties.append('summary')
|
| 232 |
-
available_classes = sorted(client.show_classes())
|
|
|
|
|
|
|
|
|
|
| 233 |
# st.write(f"Available classes: {available_classes}")
|
| 234 |
# st.write(f"Available classes type: {type(available_classes)}")
|
| 235 |
logger.info(available_classes)
|
|
@@ -324,6 +327,8 @@ def main():
|
|
| 324 |
check_model(model_name_or_path)
|
| 325 |
try:
|
| 326 |
client, available_classes = get_weaviate_client(Wapi_key, url, model_name_or_path, openai_api_key)
|
|
|
|
|
|
|
| 327 |
except Exception as e:
|
| 328 |
# Weaviate doesn't know this model, maybe we're just finetuning a model
|
| 329 |
st.sidebar.write(f"Model unknown to Weaviate")
|
|
@@ -334,8 +339,8 @@ def main():
|
|
| 334 |
class_name = st.selectbox(
|
| 335 |
label='Class Name:',
|
| 336 |
options=available_classes,
|
| 337 |
-
index=available_classes.index(start_class),
|
| 338 |
-
placeholder='Select Class Name'
|
| 339 |
)
|
| 340 |
|
| 341 |
st.write("----------")
|
|
@@ -368,7 +373,6 @@ def main():
|
|
| 368 |
st.text(f"Hybrid MRR: {vector_mrr}")
|
| 369 |
st.text(f"Total misses: {total_misses}")
|
| 370 |
|
| 371 |
-
|
| 372 |
st.write("----------")
|
| 373 |
|
| 374 |
st.title("Chat with the Impact Theory podcasts!")
|
|
|
|
| 229 |
model_name_or_path=model_name_or_path,
|
| 230 |
openai_api_key=openai_api_key)
|
| 231 |
client.display_properties.append('summary')
|
| 232 |
+
# available_classes = sorted(client.show_classes()) # doesn't work anymore
|
| 233 |
+
# print(available_classes)
|
| 234 |
+
available_classes = sorted([c['class'] for c in client.schema.get()['classes']])
|
| 235 |
+
# print(available_classes)
|
| 236 |
# st.write(f"Available classes: {available_classes}")
|
| 237 |
# st.write(f"Available classes type: {type(available_classes)}")
|
| 238 |
logger.info(available_classes)
|
|
|
|
| 327 |
check_model(model_name_or_path)
|
| 328 |
try:
|
| 329 |
client, available_classes = get_weaviate_client(Wapi_key, url, model_name_or_path, openai_api_key)
|
| 330 |
+
print(available_classes)
|
| 331 |
+
|
| 332 |
except Exception as e:
|
| 333 |
# Weaviate doesn't know this model, maybe we're just finetuning a model
|
| 334 |
st.sidebar.write(f"Model unknown to Weaviate")
|
|
|
|
| 339 |
class_name = st.selectbox(
|
| 340 |
label='Class Name:',
|
| 341 |
options=available_classes,
|
| 342 |
+
index=available_classes.index(start_class),
|
| 343 |
+
placeholder='Select Class Name'
|
| 344 |
)
|
| 345 |
|
| 346 |
st.write("----------")
|
|
|
|
| 373 |
st.text(f"Hybrid MRR: {vector_mrr}")
|
| 374 |
st.text(f"Total misses: {total_misses}")
|
| 375 |
|
|
|
|
| 376 |
st.write("----------")
|
| 377 |
|
| 378 |
st.title("Chat with the Impact Theory podcasts!")
|