Update main.py
Browse files
main.py
CHANGED
|
@@ -159,8 +159,6 @@ def getChat(text,model,tokenizer):
|
|
| 159 |
|
| 160 |
app = FastAPI()
|
| 161 |
|
| 162 |
-
pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
|
| 163 |
-
|
| 164 |
def model_init():
|
| 165 |
# args = set_args()
|
| 166 |
acuda = torch.cuda.is_available() and not args.no_cuda
|
|
|
|
| 159 |
|
| 160 |
app = FastAPI()
|
| 161 |
|
|
|
|
|
|
|
| 162 |
def model_init():
|
| 163 |
# args = set_args()
|
| 164 |
acuda = torch.cuda.is_available() and not args.no_cuda
|