Added Project files

app.py

@@ -7,7 +7,7 @@ from model import create_vit_b16
 
 # loading the classnames
 class_names = []
-with open("/content/Birds_Classification/class_names.txt", "r") as f:
+with open("class_names.txt", "r") as f:
   for cls in f.readlines():
     class_names.append(cls[:-1])
 
@@ -35,7 +35,7 @@ def predict(img):
     preds = torch.softmax(pred_logits, dim=1)
     
     # Create a prediction label and prediction probability dictionary for each prediction class
-    pred_and_prob_labels = {class_names[i]: preds[i]*100 for i in range(len(class_names))}
+    pred_and_prob_labels = {class_names[i]: preds[i].item() for i in range(len(class_names))}
 
   return pred_and_prob_labels