Update read.py

read.py

@@ -1,13 +1,9 @@
-import random
 import torch
 from PIL import Image
-import requests
 
-def classify(model, imgPath, trans=None, classes=[], device=torch.device("cpu")):
-    
+def classify(model, img, trans=None, classes=[], device=torch.device("cpu")):
     try:
         model = model.eval()
-        img = imgPath
         img = img.convert("RGB")
         img = trans(img)
         img = img.unsqueeze(0)
@@ -19,10 +15,4 @@ def classify(model, imgPath, trans=None, classes=[], device=torch.device("cpu"))
         
         return f"It {classes[pred.item()]} i'm {procent[0][pred[0]]*100:.2f}% sure"
     except Exception:
-        return "Something went wrong😕, please notify the developer with the following message: " + str(Exception)
-
-
-def get_random_quote():
-    with open("quotes.txt", "r") as file:
-        quotes = file.readlines()
-    return quotes[random.randint(0, len(quotes)-1)]
+        return "Something went wrong😕, please notify the developer with the following message: " + str(Exception)