Hexii commited on
Commit
d96954f
1 Parent(s): 6e8867f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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 [12 different types](https://huggingface.co/spaces/Hexii/Cat-Breed-Classifier/blob/main/classes.txt)"
38
- article = "<p style='text-align: center'><a href='https://github.com/Mr-Hexi' target='_blank'>GitHub</a>![visitors](https://visitor-badge.glitch.me/badge?page_id=Hexii.Cat-Breed-Classifier)</p> "
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)