heikowagner commited on
Commit
55c987d
1 Parent(s): 515fe55

Update app/run.py

Browse files
Files changed (1) hide show
  1. 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
- #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
- #llm= load_model.load_cpu_model()
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
  # %%