Update app.py

app.py

@@ -1,29 +1,9 @@
-# !pip install torch
-# import torch
 from transformers import AutoTokenizer, AutoModel ,AutoConfig
 import torch
 from transformers import  ViTImageProcessor, VisionEncoderDecoderModel,RobertaTokenizerFast
 import PIL 
-
-
-# Move model to GPU , depnding on device
-device2 = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-# Load the model
-
-
-
 import streamlit as st
 from PIL import Image
-# from transformers import GPT2TokenizerFast, ViTImageProcessor, VisionEncoderDecoderModel,RobertaTokenizerFast, VisionEncoderDecoderModel
-#from transformers import BlipProcessor, BlipForConditionalGeneration
-
-
-# Load model directly
-from transformers import AutoTokenizer, AutoModel
-
-# tokenizer = AutoTokenizer.from_pretrained("sourabhbargi11/Caption_generator_model")
-# model = AutoModel.from_pretrained("sourabhbargi11/Caption_generator_model")
-
 
 
 def set_page_config():
@@ -93,7 +73,6 @@ def main():
 if __name__ == '__main__':
     main()
 
-
 # st.markdown("""
 #     ---
 #    You are looking at partial tuned model , please JUDGE ME!!!  (I am Funny , Sensible , Creative )""")
@@ -101,5 +80,3 @@ if __name__ == '__main__':
 st.markdown("""
     ---
    You are looking at a partially tuned model. Judge me! (I am Funny and Creative) 😄🎨""")
-
-