Hugging Face's logo

Spaces:
heikowagner
/
GPT-Docker
Build error

App Files Community
4
heikowagner commited on
Commit
f02eb64
•
1 Parent(s): 1f3a255

Update app/run.py

Browse files
Files changed (1) hide show
  1. app/run.py +1 -1
app/run.py CHANGED
@@ -46,7 +46,7 @@ create_and_add(collection_name, sub_docs, model_name, metadata)
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)
 
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, k=1)
50
  result = chain({"query": "Ist mein Kinderwagen bei einem Leitungswasserschaden mitversichert?"})
51
  print(result)
52
  #llm= load_model.load_openai_model(temperature=0.1)