Update app.py

app.py

@@ -14,6 +14,7 @@ repo_dir = snapshot_download(repo_id=MODEL_REPO_ID)
 
 # Load predictor directly
 PREDICTOR = MultiModalPredictor.load(repo_dir)
+TMP_PATH = pathlib.Path("tmp_input.png")
 
 # Explicit class labels
 CLASS_LABELS = {0: "🚫 No Stop Sign", 1: "🛑 Stop Sign"}
@@ -24,9 +25,7 @@ def do_predict(pil_img: PIL.Image.Image):
     pil_resized = pil_img.convert("RGB").resize((256, 256))
     
     # Save to tmp for AutoGluon input
-    tmpdir = pathlib.Path(tempfile.mkdtemp())
-    img_path = tmpdir / "input.png"
-    pil_img.save(img_path)
+    pil_resized.save(TMP_PATH)
 
     df = pd.DataFrame({"image": [str(img_path)]})