ChrisMofus commited on
Commit
7365cb7
1 Parent(s): 1a7e620
Files changed (2) hide show
  1. app.py +1 -0
  2. requirements.txt +1 -0
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import streamlit as st
2
  from PIL import Image
 
3
 
4
  from transformers import AutoProcessor, AutoModelForVision2Seq
5
 
 
1
  import streamlit as st
2
  from PIL import Image
3
+ import torch
4
 
5
  from transformers import AutoProcessor, AutoModelForVision2Seq
6
 
requirements.txt CHANGED
@@ -1,3 +1,4 @@
1
  streamlit
2
  Pillow
 
3
  transformers
 
1
  streamlit
2
  Pillow
3
+ torch
4
  transformers