obx0x3 commited on
Commit
7dabbf8
·
verified ·
1 Parent(s): 54b6cf1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -6,7 +6,6 @@ import uvicorn
6
 
7
  app = FastAPI()
8
 
9
- # Load fine-tuned model from local path
10
  tokenizer = T5Tokenizer.from_pretrained("obx0x3/empathy-dementia")
11
  model = T5ForConditionalGeneration.from_pretrained("obx0x3/empathy-dementia")
12
 
@@ -30,4 +29,4 @@ async def generate_response(payload: PromptRequest):
30
  return {"reply": result}
31
 
32
  if __name__ == "__main__":
33
- uvicorn.run(app, host="0.0.0.0", port=7860)
 
6
 
7
  app = FastAPI()
8
 
 
9
  tokenizer = T5Tokenizer.from_pretrained("obx0x3/empathy-dementia")
10
  model = T5ForConditionalGeneration.from_pretrained("obx0x3/empathy-dementia")
11
 
 
29
  return {"reply": result}
30
 
31
  if __name__ == "__main__":
32
+ uvicorn.run(app, host="0.0.0.0", port=7860)