Spaces:
Build error
Build error
heikowagner
commited on
Commit
•
55c987d
1
Parent(s):
515fe55
Update app/run.py
Browse files- app/run.py +5 -5
app/run.py
CHANGED
@@ -43,14 +43,14 @@ sub_docs = load_and_split(docs, chunk_size=1000)
|
|
43 |
create_and_add(collection_name, sub_docs, model_name, metadata)
|
44 |
|
45 |
|
46 |
-
|
47 |
# %%
|
48 |
-
|
49 |
-
|
50 |
-
|
51 |
#llm= load_model.load_openai_model(temperature=0.1)
|
52 |
|
53 |
-
|
54 |
|
55 |
#meta= metadata_generator(docs[0], llm)
|
56 |
# %%
|
|
|
43 |
create_and_add(collection_name, sub_docs, model_name, metadata)
|
44 |
|
45 |
|
46 |
+
llm= load_model.load_cpu_model()
|
47 |
# %%
|
48 |
+
chain = load_model.create_chain(llm, collection=collection_name, model_name=model_name, metadata=metadata)
|
49 |
+
result = chain({"query": "Ist mein Kinderwagen bei einem Leitungswasserschaden mitversichert?"})
|
50 |
+
print(result)
|
51 |
#llm= load_model.load_openai_model(temperature=0.1)
|
52 |
|
53 |
+
|
54 |
|
55 |
#meta= metadata_generator(docs[0], llm)
|
56 |
# %%
|