Update app.py

app.py

@@ -24,7 +24,7 @@ def process_image(image):
     expression_text = tokenizer(expression_labels)
     clothing_text = tokenizer(clothing_labels)
 
-    with torch.no_grad(), torch.cuda.amp.autocast():
+    with torch.no_grad():
         # Encode image and text
         image_features = model.encode_image(image)
         type_text_features = model.encode_text(type_text)
@@ -58,10 +58,10 @@ iface = gr.Interface(
     fn=process_image,
     inputs=gr.Image(type="pil", label="Upload Image"),
     outputs=[
-        gr.Label(label="Type Classification", num_top_classes=3),
-        gr.Label(label="Scene Classification", num_top_classes=3),
-        gr.Label(label="Expression Classification", num_top_classes=3),
-        gr.Label(label="Clothing Classification", num_top_classes=3)
+        gr.Label(label="Type Classification", num_top_classes=8),
+        gr.Label(label="Scene Classification", num_top_classes=8),
+        gr.Label(label="Expression Classification", num_top_classes=8),
+        gr.Label(label="Clothing Classification", num_top_classes=8)
     ],
     title="Image Content Moderation",
     description="Upload an image to analyze its content across multiple categories."