Hugging Face's logo

Spaces:
AmandaPanda
/
Wk8Assignment_01
Sleeping

App Files Community
AmandaPanda commited on
Commit
3f77bf2
·
verified ·
1 Parent(s): 607f6c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -36,9 +36,11 @@ processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-larg
36
  model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large")
37
 
38
  #Load transformer for translating captions from English to Spanish
39
- trans_tokenizer = AutoTokenizer.from_pretrained("Abhra-loony/english-to-spanish-lang-translation-model")
40
  trans_model = AutoModelForSeq2SeqLM.from_pretrained("Abhra-loony/english-to-spanish-lang-translation-model")
41
 
 
 
42
  #Configure captioning function
43
  def caption_random_image():
44
 
 
36
  model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large")
37
 
38
  #Load transformer for translating captions from English to Spanish
39
+ trans_tokenizer = AutoTokenizer.from_pretrained("Abhra-loony/english-to-spanish-lang-translation-model", from_tf=True)
40
  trans_model = AutoModelForSeq2SeqLM.from_pretrained("Abhra-loony/english-to-spanish-lang-translation-model")
41
 
42
+
43
+
44
  #Configure captioning function
45
  def caption_random_image():
46