Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ effnetb2, effnetb2_transforms = create_effnetb2_model(
|
|
17 |
effnetb2.load_state_dict(
|
18 |
torch.load(
|
19 |
f="Cat_Breed_Classifier_12_class_90_acc.pth",
|
20 |
-
map_location=torch.device("cpu"),
|
21 |
)
|
22 |
)
|
23 |
|
@@ -33,9 +33,9 @@ def predict(img):
|
|
33 |
pred_time = round(timer() - start_time, 5)
|
34 |
return pred_labels_and_probs, pred_time
|
35 |
|
36 |
-
title = "Cat Breed Classifier Demo
|
37 |
-
description = "Gradio Demo for Classifying Cat Breeds of these
|
38 |
-
article = "
|
39 |
|
40 |
|
41 |
|
@@ -54,4 +54,4 @@ app = gr.Interface(
|
|
54 |
article=article,
|
55 |
)
|
56 |
|
57 |
-
app.launch()
|
|
|
17 |
effnetb2.load_state_dict(
|
18 |
torch.load(
|
19 |
f="Cat_Breed_Classifier_12_class_90_acc.pth",
|
20 |
+
map_location=torch.device("cpu"), # load to CPU
|
21 |
)
|
22 |
)
|
23 |
|
|
|
33 |
pred_time = round(timer() - start_time, 5)
|
34 |
return pred_labels_and_probs, pred_time
|
35 |
|
36 |
+
title = "Cat Breed Classifier Demo 😸"
|
37 |
+
description = "<p style='text-align: center'>Gradio Demo for Classifying Cat Breeds of these <a href='https://huggingface.co/'>5 different types.<a></p>"
|
38 |
+
article = "</br><p style='text-align: center'><a href='https://github.com/Mr-Hexi' target='_blank'>GitHub</a></br>![visitors](https://visitor-badge.glitch.me/badge?page_id=Hexii.Cat-Breed-Classifier)</p> "
|
39 |
|
40 |
|
41 |
|
|
|
54 |
article=article,
|
55 |
)
|
56 |
|
57 |
+
app.launch(share=True)
|