sourabhbargi11 commited on
Commit
1531e43
·
verified ·
1 Parent(s): b26fc84

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -84,7 +84,6 @@ def main():
84
  if st.sidebar.button('Generate Caption'):
85
  with st.spinner('Generating caption...'):
86
  image_processor, model,tokenizer, device = initialize_model()
87
-
88
  caption = generate_caption(image_processor, model, device, image)
89
 
90
  st.header("Caption:")
@@ -97,5 +96,5 @@ if __name__ == '__main__':
97
 
98
  st.markdown("""
99
  ---
100
- You are looking at partial finetuned model , please JUDGE ME!!! """)
101
 
 
84
  if st.sidebar.button('Generate Caption'):
85
  with st.spinner('Generating caption...'):
86
  image_processor, model,tokenizer, device = initialize_model()
 
87
  caption = generate_caption(image_processor, model, device, image)
88
 
89
  st.header("Caption:")
 
96
 
97
  st.markdown("""
98
  ---
99
+ You are looking at partial tuned model , please JUDGE ME!!! (I am Funny , Sensible , Creative )""")
100