No Troll

app.py

@@ -27,15 +27,15 @@ button:hover {
 """
 
 def classify_img(img) -> dict:
-	"""helper function to generate predictions"""
-	
-	cats = {'Dog', 'Cat'}
-	preds, idx, probs = learn.predict(img)
-	
-	return dict(zip(cats, map(float, probs)))
+    """helper function to generate predictions"""
+    
+    lbls = {'Cat', 'Dog'}
+    preds, idx, probs = model.predict(img)
+    
+    return dict(zip(lbls, map(float, probs)))
 
 # load model
-learn = load_learner('model_2_ep_new_data.pkl')
+model = load_learner('model_2_ep_new_data.pkl')
 
 # build app
 image = gr.Image()

requirements.txt

@@ -1,3 +1,5 @@
 fastai==2.7.18
 torch==2.5.1
-torchvision==0.20.1
+torchvision==0.20.1
+gradio==5.15.0
+gradio_client==1.7.0