Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
from fastapi import FastAPI
|
| 2 |
from pydantic import BaseModel
|
| 3 |
from typing import Optional
|
| 4 |
-
from transformers import pipeline,
|
| 5 |
from fastapi.middleware.cors import CORSMiddleware
|
| 6 |
from datasets import load_dataset
|
| 7 |
from huggingface_hub import login
|
|
|
|
| 1 |
from fastapi import FastAPI
|
| 2 |
from pydantic import BaseModel
|
| 3 |
from typing import Optional
|
| 4 |
+
from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
|
| 5 |
from fastapi.middleware.cors import CORSMiddleware
|
| 6 |
from datasets import load_dataset
|
| 7 |
from huggingface_hub import login
|