out_string

camenduru · Feb 3, 2024 · 89040d1 · 89040d1
1 parent 6849ceb
commit 89040d1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/predict.py b/predict.py
@@ -50,6 +50,6 @@ def predict(
         self,
         input_image: Path = Input(description="Input Image"),
         input_text: str = Input(default="What is unusual about this image?"),
-    ) -> Path:
-        video_path = inference(input_image, input_text, self.tokenizer, self.model, self.processor)
-        return Path(video_path)
+    ) -> str:
+        out_string = inference(input_image, input_text, self.tokenizer, self.model, self.processor)
+        return out_string