Cat-Dog-Classifier

Sleeping

JaaackXD commited on Jun 29

Commit

e462381

•

1 Parent(s): afa24f1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -63,8 +63,8 @@ def inference(input_image):
         result[model_categories[top1_catid[i]]] = top1_prob[i].item()
     return result
-inputs = gr.inputs.Image(type='pil')
-outputs = gr.outputs.Label(type="confidences", num_top_classes = n_categories)
 title = "STAT 430 Final Project App -- Made by Group DHZ"
 description = "This is our Cat & Dog Classifier for the final project, and the model we use is generated by our second neural network augmented by the flipping technique, which is would give the best accuracy. To use it, simply upload your image, or click one of the examples to load them. The authors are Xiongjie Dai (xdai12), Yu He (yuhe6), Mengjia Zeng (mengjia6)."

         result[model_categories[top1_catid[i]]] = top1_prob[i].item()
     return result
+inputs = gr.Image(type='pil')
+outputs = gr.Label(type="confidences", num_top_classes=n_categories)
 title = "STAT 430 Final Project App -- Made by Group DHZ"
 description = "This is our Cat & Dog Classifier for the final project, and the model we use is generated by our second neural network augmented by the flipping technique, which is would give the best accuracy. To use it, simply upload your image, or click one of the examples to load them. The authors are Xiongjie Dai (xdai12), Yu He (yuhe6), Mengjia Zeng (mengjia6)."