Spaces:
Runtime error
Runtime error
Topallaj Denis commited on
Commit ·
6aadb65
1
Parent(s): 94e262e
allow models to be used
Browse files
main.py
CHANGED
|
@@ -14,13 +14,11 @@ import pydantic
|
|
| 14 |
|
| 15 |
app = FastAPI()
|
| 16 |
|
| 17 |
-
"""
|
| 18 |
tokenizer = T5Tokenizer.from_pretrained(
|
| 19 |
"Rostlab/prot_t5_xl_half_uniref50-enc", do_lower_case=False, torch_dtype=torch.float16)
|
| 20 |
|
| 21 |
model = T5EncoderModel.from_pretrained(
|
| 22 |
"Rostlab/prot_t5_xl_half_uniref50-enc")
|
| 23 |
-
"""
|
| 24 |
|
| 25 |
class Item(pydantic.BaseModel):
|
| 26 |
sequence: str
|
|
|
|
| 14 |
|
| 15 |
app = FastAPI()
|
| 16 |
|
|
|
|
| 17 |
tokenizer = T5Tokenizer.from_pretrained(
|
| 18 |
"Rostlab/prot_t5_xl_half_uniref50-enc", do_lower_case=False, torch_dtype=torch.float16)
|
| 19 |
|
| 20 |
model = T5EncoderModel.from_pretrained(
|
| 21 |
"Rostlab/prot_t5_xl_half_uniref50-enc")
|
|
|
|
| 22 |
|
| 23 |
class Item(pydantic.BaseModel):
|
| 24 |
sequence: str
|