Update app.py

app.py

@@ -6,7 +6,7 @@ from transformers import (
     AutoModelForImageClassification
 )
 
-pre_process = AutoImageProcessor.from_pretrained('javierrf91/streamlit')
+pre_process = AutoImageProcessor.from_pretrained('adwod/ViT_bluetab')
 
 
 st.title("What is it?")
@@ -20,6 +20,6 @@ if file_name is not None:
     col1.image(image, use_column_width=True)
     inputs = pre_process(images=image, return_tensors="pt")
     input_pixels = inputs.pixel_values
-    model = AutoModelForImageClassification.from_pretrained('javierrf91/streamlit')
+    model = AutoModelForImageClassification.from_pretrained('adwod/ViT_bluetab'')
     outputs = model(input_pixels)
     col2.header(model.config.id2label[outputs.logits.argmax(-1)  .item()])