Update app.py

app.py

@@ -5,20 +5,30 @@ def gpu():
     
 import os
 import subprocess
+import sys
 import gradio as gr
 
 MODEL_PATH = os.environ.get("MODEL_PATH", "models/BitNet-b1.58-2B-4T/ggml-model-i2_s.gguf")
 
 def generate(prompt, max_tokens=128, temperature=0.7):
+    cwd = os.path.dirname(__file__)
     cmd = [
-        "python", "run_inference.py",
+        sys.executable,
+        os.path.join(cwd, "run_inference.py"),
         "-m", MODEL_PATH,
         "-p", prompt,
         "-n", str(max_tokens),
         "-temp", str(temperature)
     ]
-    result = subprocess.run(cmd, capture_output=True, text=True)
-    return result.stdout.strip()
+    proc = subprocess.run(
+        cmd,
+        capture_output=True,
+        text=True,
+        cwd=cwd
+    )
+    if proc.returncode != 0:
+        return f"error: {proc.stderr.strip()}"
+    return proc.stdout.strip()
 
 iface = gr.Interface(
     fn=generate,