Added requirements to work locally, and changed channel for img out

app.py

@@ -143,7 +143,11 @@ def inference(query, labels, n_supp=10,
         fig.canvas.draw()
         w, h = fig.canvas.get_width_height()
         img = np.frombuffer(fig.canvas.tostring_argb(), dtype=np.uint8).reshape(h, w, 4)
-        img = img[:, :, :3]
+        
+        # The image is all red, we are probably not cutting the alpha channel
+        #img = img[:, :, :3]
+        img = img[:, :, 1:4] # remove alpha channel
+        
         return {k: float(v) for k, v in zip(labels, probs[0, 0])}, Image.fromarray(img)
         #return {k: float(v) for k, v in zip(labels, probs[0, 0])}, fig
 

requirements.txt

@@ -10,3 +10,5 @@ ml-collections
 ftfy
 tensorboard
 Google-Images-Search
+semantic-version
+pytz