kouki321 commited on
Commit
eefa3ef
·
verified ·
1 Parent(s): e4566db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import os
2
  import torch
 
3
  from fastapi import FastAPI, File, UploadFile, HTTPException, Body
4
  from fastapi.responses import JSONResponse
5
  from transformers import AutoTokenizer, AutoModelForCausalLM
 
1
  import os
2
  import torch
3
+ import uvicorn
4
  from fastapi import FastAPI, File, UploadFile, HTTPException, Body
5
  from fastapi.responses import JSONResponse
6
  from transformers import AutoTokenizer, AutoModelForCausalLM