D0k-tor commited on
Commit
2baef79
·
1 Parent(s): 79fef48

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,7 +1,9 @@
1
  import torch
2
  import re
3
  import gradio as gr
4
- import streamlit as st
 
 
5
  # st.title("Image Caption Generator")
6
  from transformers import AutoTokenizer, ViTFeatureExtractor, VisionEncoderDecoderModel
7
  import os
 
1
  import torch
2
  import re
3
  import gradio as gr
4
+ # import streamlit as st
5
+ from PIL import Image
6
+
7
  # st.title("Image Caption Generator")
8
  from transformers import AutoTokenizer, ViTFeatureExtractor, VisionEncoderDecoderModel
9
  import os