Update prediction.py

prediction.py

@@ -19,10 +19,10 @@ class smartcities:
         detect_fn = tf.saved_model.load("model/saved_model")
         self.detect_fn = detect_fn
 
-    def predict(self):
+    def predict(self, input):
         # Ruta del video (Se debe cargar de manera manual)
-        PATH_VIDEO = "/tmp/in_video.mp4"
-        video_result = open(PATH_VIDEO, "wb")
+        # PATH_VIDEO = "/tmp/in_video.mp4"
+        # video_result = open(PATH_VIDEO, "wb")
         # video_result.write(base64.b64decode(image_64_decode))
 
         # Ruta del video en donde almacenaremos los resultados
@@ -35,7 +35,7 @@ class smartcities:
         TRESHOLD = 0.5
 
         # Cargamos el video
-        vs = cv2.VideoCapture(PATH_VIDEO)
+        vs = cv2.VideoCapture(input)
 
         # Inicializamos el writer para poder guardar el video
         writer = None