Debugin code

app.py

@@ -4,16 +4,14 @@ from git import Repo
 Repo.clone_from("https://github.com/dimitreOliveira/hub.git", "./hub")
 sys.path.append("/hub")
 
-import requests
-# Download human-readable labels for ImageNet.
-response = requests.get("https://storage.googleapis.com/download.tensorflow.org/data/ImageNetLabels.txt")
-labels = [x for x in response.text.split("\n") if x != ""]
-
-
 import gradio as gr
-import tensorflow as tf
 from hub.tensorflow_hub.hf_utils import pull_from_hub
 
+# import requests
+# # Download human-readable labels for ImageNet.
+# response = requests.get("https://storage.googleapis.com/download.tensorflow.org/data/ImageNetLabels.txt")
+# labels = [x for x in response.text.split("\n") if x != ""]
+
 model = pull_from_hub(repo_id="Dimitre/mobilenet_v3_small")
 
 def preprocess(image):
@@ -25,14 +23,14 @@ def preprocess(image):
     print(image / 255.)
     return image / 255.
 
-def postprocess(prediction):
-    return {labels[i]: prediction[i] for i in range(len(labels))}
+# def postprocess(prediction):
+#     return {labels[i]: prediction[i] for i in range(len(labels))}
 
 def predict_fn(image):
     image = preprocess(image)
     prediction = model([image])
-    scores = postprocess(prediction)
-    return scores
+    # scores = postprocess(prediction)
+    return prediction
 
 iface = gr.Interface(fn=predict_fn, 
                      inputs=gr.Image(shape=(224, 224)), 

requirements.txt

@@ -1,2 +1 @@
-GitPython
-tensorflow
+GitPython