Yair commited on
Commit
1fb4187
1 Parent(s): 6437a53

add requirements.txt

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. requirements.txt +3 -0
app.py CHANGED
@@ -19,6 +19,7 @@ def create_caption_transformer(img):
19
  return caption_text
20
 
21
 
 
22
  IMAGES_EXAMPLES_FOLDER = 'examples/'
23
  images = os.listdir(IMAGES_EXAMPLES_FOLDER)
24
  IMAGES_EXAMPLES = [IMAGES_EXAMPLES_FOLDER + img for img in images]
19
  return caption_text
20
 
21
 
22
+ torch.__version__
23
  IMAGES_EXAMPLES_FOLDER = 'examples/'
24
  images = os.listdir(IMAGES_EXAMPLES_FOLDER)
25
  IMAGES_EXAMPLES = [IMAGES_EXAMPLES_FOLDER + img for img in images]
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ torch~=1.12.0
2
+ gradio~=3.0.22
3
+ transformers~=4.20.1