heikowagner commited on
Commit
3ad0959
1 Parent(s): de3c0e5

Update app/run.py

Browse files
Files changed (1) hide show
  1. app/run.py +4 -3
app/run.py CHANGED
@@ -45,9 +45,10 @@ create_and_add(collection_name, sub_docs, model_name, metadata)
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()
 
45
 
46
 
47
  # %%
48
+ llm= load_model.load_cpu_model()
49
+ chain = load_model.create_chain(llm, collection=collection_name, model_name=model_name, metadata=metadata)
50
+ result = chain({"query": "Ist mein Kinderwagen bei einem Leitungswasserschaden mitversichert?"})
51
+ print(result)
52
  #llm= load_model.load_openai_model(temperature=0.1)
53
 
54
  #llm= load_model.load_cpu_model()