test path

app.py

@@ -15,6 +15,7 @@ import time
 from safetensors.torch import load_file
 import time
 import tempfile
+from pathlib import Path
 
 # Constants
 BASE = "stabilityai/stable-diffusion-xl-base-1.0"
@@ -136,7 +137,7 @@ def predict(prompt, seed=1231231):
     image = results.images[0]
     with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as tmpfile:
         image.save(tmpfile, "JPEG", quality=80, optimize=True, progressive=True)
-        return tmpfile.name
+        return Path(tmpfile.name)
 
 
 css = """