Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from transformers import CLIPModel, CLIPProcessor, pipeline
+from transformers import CLIPModel, pipeline, CLIPImageProcessor 
 import torch
 from PIL import Image
 
@@ -13,7 +13,7 @@ def load_clip(model_size='large'):
         MODEL_name = 'openai/clip-vit-large-patch14'
     
     model = CLIPModel.from_pretrained(MODEL_name)
-    processor = CLIPProcessor.from_pretrained(MODEL_name)
+    processor = CLIPImageProcessor.from_pretrained(MODEL_name)
 
     return processor, model
 
@@ -48,11 +48,11 @@ with col_l:
 
 if picture_file is not None:
     image = Image.open(picture_file)
-    st.image(image, caption='Please upload an image of the damage', use_column_width=True)
+    st.image(image, caption='Please upload an image of the damage') #use_column_width=True
 
 #image
 with col_l:
-    default_options = ['black', 'white', 'gray', 'red', 'blue', 'silver', 'red', 'brown', 'green', 'orange', 'beige', 'pruple', 'gold', 'yellow']
+    default_options = ['There is a car', 'There is no car']
     options = st.text_input(label="Please enter the classes", value=default_options)
     #options = list(options)