rajeshradhakrishnan commited on
Commit
a894f41
1 Parent(s): 696bdbb

update to facebook Bart50-v1

Browse files
Files changed (2) hide show
  1. main.py +1 -1
  2. 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 pipeline, MBartForConditionalGeneration, MBart50TokenizerFast
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.*