Hugging Face's logo

Spaces:
sourabhbargi11
/
caption_generate
Sleeping

App Files Community
sourabhbargi11 commited on
Commit
c749f9d
·
verified ·
1 Parent(s): 0610d82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,10 +1,10 @@
 
1
 
2
- !pip install transformers
3
  import streamlit as st
4
  from PIL import Image
5
  from transformers import GPT2TokenizerFast, ViTImageProcessor, VisionEncoderDecoderModel,RobertaTokenizerFast, VisionEncoderDecoderModel
6
 
7
- #import torch
8
  #from transformers import BlipProcessor, BlipForConditionalGeneration
9
 
10
  def set_page_config():
 
1
+ import torch
2
 
 
3
  import streamlit as st
4
  from PIL import Image
5
  from transformers import GPT2TokenizerFast, ViTImageProcessor, VisionEncoderDecoderModel,RobertaTokenizerFast, VisionEncoderDecoderModel
6
 
7
+
8
  #from transformers import BlipProcessor, BlipForConditionalGeneration
9
 
10
  def set_page_config():