Spaces:
Runtime error
Runtime error
rajeshradhakrishnan
commited on
Commit
•
a894f41
1
Parent(s):
696bdbb
update to facebook Bart50-v1
Browse files- main.py +1 -1
- requirements.txt +1 -0
main.py
CHANGED
@@ -2,7 +2,7 @@ import os
|
|
2 |
from fastapi import FastAPI, Request
|
3 |
from fastapi.staticfiles import StaticFiles
|
4 |
from fastapi.responses import FileResponse
|
5 |
-
from transformers import
|
6 |
|
7 |
|
8 |
model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-one-to-many-mmt")
|
|
|
2 |
from fastapi import FastAPI, Request
|
3 |
from fastapi.staticfiles import StaticFiles
|
4 |
from fastapi.responses import FileResponse
|
5 |
+
from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
|
6 |
|
7 |
|
8 |
model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-one-to-many-mmt")
|
requirements.txt
CHANGED
@@ -3,4 +3,5 @@ requests==2.27.*
|
|
3 |
sentencepiece==0.1.*
|
4 |
torch==1.11.*
|
5 |
transformers==4.*
|
|
|
6 |
uvicorn[standard]==0.17.*
|
|
|
3 |
sentencepiece==0.1.*
|
4 |
torch==1.11.*
|
5 |
transformers==4.*
|
6 |
+
protobuf
|
7 |
uvicorn[standard]==0.17.*
|