Update app.py

app.py

@@ -85,7 +85,7 @@ transform = transforms.Compose([
 
 
 def predict(inp, use_retrieval, use_shape_classification, use_period_classification, use_drawing):
-    image_tensor = transform(inp).to(device)
+    image_tensor = transform(inp)
     feature = embedding_model_test(image_tensor.unsqueeze(0)) if use_retrieval else None
     results = {}