Spaces:

Vvaann
/

Assignment_13-RESNET

Sleeping

Vvaann commited on May 11, 2024

Commit

52c6c75

verified ·

1 Parent(s): d0c5619

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -64,7 +64,7 @@ demo = gr.Interface(
     ],
     outputs = [
         "text",
-        gr.IMage(width= 256, height=256,label="Output"),
         gr.Label(num_top_classes=3)
     ],
     title = "CIFAR 10 trained on ResNet model in pytorch lightning with Gradcam",

     ],
     outputs = [
         "text",
+        gr.Image(width= 256, height=256,label="Output"),
         gr.Label(num_top_classes=3)
     ],
     title = "CIFAR 10 trained on ResNet model in pytorch lightning with Gradcam",