Update app.py

app.py

@@ -88,8 +88,7 @@ def classify_image(image, candidate_labels, model_name):
     text_tokens = tokenizer(labels)
     
     # Run inference
-    # with torch.no_grad(), torch.cuda.amp.autocast():
-    with torch.no_grad():
+    with torch.no_grad(), torch.cuda.amp.autocast():
         image_features = model.encode_image(image_tensor)
         text_features = model.encode_text(text_tokens)