CR7CAD commited on
Commit
3b82d8f
·
verified ·
1 Parent(s): 7c4bc18

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,7 +1,8 @@
1
- # Only the two imports you requested
2
  import streamlit as st
3
  from transformers import pipeline
4
  from PIL import Image
 
5
 
6
  # Simple image-to-text function
7
  def img2text(image):
 
1
+ # Imports
2
  import streamlit as st
3
  from transformers import pipeline
4
  from PIL import Image
5
+ import torch
6
 
7
  # Simple image-to-text function
8
  def img2text(image):