Spaces:

khan994
/

draft

Sleeping

khan994 commited on Feb 6, 2023

Commit

2579041

1 Parent(s): 6cb26de

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -57,7 +57,7 @@ def gradcam(img_create):
   im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB)
   #im=np.fromstring(fig.canvas.tostring_rgb(), dtype=np.uint8, sep='')
-  return dict(zip(categories, map(float, probs))), im
 categories = ('arbanasi', 'filibe', 'gjirokoster', 'iskodra', 'kula', 'kuzguncuk', 'larissa_ampelakia', 'mardin', 'ohrid', 'pristina', 'safranbolu', 'selanik', 'sozopol_suzebolu', 'tiran', 'varna')
 def classify_img(img):

   im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB)
   #im=np.fromstring(fig.canvas.tostring_rgb(), dtype=np.uint8, sep='')
+  return im, dict(zip(categories, map(float, probs)))
 categories = ('arbanasi', 'filibe', 'gjirokoster', 'iskodra', 'kula', 'kuzguncuk', 'larissa_ampelakia', 'mardin', 'ohrid', 'pristina', 'safranbolu', 'selanik', 'sozopol_suzebolu', 'tiran', 'varna')
 def classify_img(img):