ChrisMofus commited on
Commit
1a7e620
1 Parent(s): ed6a975

Add dependencies

Browse files
Files changed (2) hide show
  1. app.py +0 -1
  2. requirements.txt +3 -1
app.py CHANGED
@@ -2,7 +2,6 @@ import streamlit as st
2
  from PIL import Image
3
 
4
  from transformers import AutoProcessor, AutoModelForVision2Seq
5
- from transformers.image_utils import load_image
6
 
7
  DEVICE = "cuda:0"
8
 
 
2
  from PIL import Image
3
 
4
  from transformers import AutoProcessor, AutoModelForVision2Seq
 
5
 
6
  DEVICE = "cuda:0"
7
 
requirements.txt CHANGED
@@ -1 +1,3 @@
1
- torch
 
 
 
1
+ streamlit
2
+ Pillow
3
+ transformers