ngdwtrg commited on
Commit
bdc9720
·
1 Parent(s): 15941be

use minilm-l12 embedding model

Browse files
Files changed (1) hide show
  1. main.py +2 -4
main.py CHANGED
@@ -23,7 +23,7 @@ app.add_middleware(
23
  allow_headers=["*"],
24
  )
25
 
26
- embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/paraphrase-MiniLM-L2-v2")
27
 
28
  @app.post("/uploadfile/")
29
  async def upload_file(file: UploadFile = File(...)):
@@ -56,6 +56,4 @@ async def query(request: QueryRequest):
56
  return {"question": request.question, "answer": answer}
57
  except Exception as e:
58
  raise HTTPException(status_code=500, detail=str(e))
59
- if __name__ == "__main__":
60
- import uvicorn
61
- uvicorn.run(app, host="0.0.0.0", port=int(os.getenv("PORT", 8000)))
 
23
  allow_headers=["*"],
24
  )
25
 
26
+ embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/paraphrase-MiniLM-L12-v2")
27
 
28
  @app.post("/uploadfile/")
29
  async def upload_file(file: UploadFile = File(...)):
 
56
  return {"question": request.question, "answer": answer}
57
  except Exception as e:
58
  raise HTTPException(status_code=500, detail=str(e))
59
+