Update app.py

app.py

@@ -14,6 +14,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
 tokenizer = AutoTokenizer.from_pretrained("rararara9999/Model")
 model = AutoModelForSequenceClassification.from_pretrained("rararara9999/Model")
+from transformers import AutoModelForImageClassification, AutoImageProcessor
 
 import torch
 import numpy as np