Spaces:
Build error
Build error
Yair
commited on
Commit
•
1fb4187
1
Parent(s):
6437a53
add requirements.txt
Browse files- app.py +1 -0
- 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
|