Update app.py

app.py

@@ -17,7 +17,7 @@ effnetb2, effnetb2_transforms = create_effnetb2_model(
 effnetb2.load_state_dict(
     torch.load(
         f="Cat_Breed_Classifier_12_class_90_acc.pth",
-        map_location=torch.device("cpu"),  # load to CPU
+        map_location=torch.device("cpu"),
     )
 )
 
@@ -33,8 +33,8 @@ def predict(img):
     pred_time = round(timer() - start_time, 5)
     return pred_labels_and_probs, pred_time
 
-title = "Cat Breed Classifier Demo 😼"
-description = "<p style='text-align: center'>Gradio Demo for  Classifying Cat Breeds of these <a href='https://huggingface.co/'>5 different types.<a></p>"
+title = "Cat Breed Classifier Demo 🐱"
+description = "<p style='text-align: center'>Gradio Demo for  Classifying Cat Breeds of these <a href='https://huggingface.co/spaces/Hexii/Cat-Breed-Classifier/blob/main/classes.txt'>12 different types.<a></p>"
 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> "