Update app.py

app.py

@@ -96,21 +96,24 @@ class ModelConverter:
         """Convert the model to ONNX format."""
         try:
             if trust_remote_code:
-                result = subprocess.run(
-                    [
-                        sys.executable,
-                        "-m",
-                        "scripts.convert",
-                        "--quantize",
-                        "--trust_remote_code",
-                        "--model_id",
-                        input_model_id,
-                    ],
-                    cwd=self.config.repo_path,
-                    capture_output=True,
-                    text=True,
-                    env={},
-                )
+                if st.session_state.get("user_hf_token") == st.secrets.get("HF_TOKEN"):
+                    result = subprocess.run(
+                        [
+                            sys.executable,
+                            "-m",
+                            "scripts.convert",
+                            "--quantize",
+                            "--trust_remote_code",
+                            "--model_id",
+                            input_model_id,
+                        ],
+                        cwd=self.config.repo_path,
+                        capture_output=True,
+                        text=True,
+                        env={},
+                    )
+                else:
+                    raise Exception("Trust Remote Code requires your own HuggingFace token.")
             else:
                 result = subprocess.run(
                     [