Update app.py

app.py

@@ -28,10 +28,9 @@ class Config:
     repo_path: Path = Path("./transformers.js")
 
     @classmethod
-    def from_env(cls) -> "Config":
+    def from_env(cls, user_token: Optional[str] = None) -> "Config":
         """Create config from environment variables and secrets."""
         system_token = os.getenv("HF_TOKEN")
-        user_token = gr.session_state.get("user_hf_token")
         if user_token:
             hf_username = whoami(token=user_token)["name"]
         else:
@@ -148,29 +147,22 @@ class ModelConverter:
 def convert_and_upload_model(input_model_id: str, user_hf_token: str = None):
     """Function to handle model conversion and upload."""
     try:
-        config = Config.from_env()
-        if user_hf_token:
-            gr.session_state["user_hf_token"] = user_hf_token
-            config = Config.from_env()
-
+        config = Config.from_env(user_hf_token)
         converter = ModelConverter(config)
         converter.setup_repository()
 
         if converter.api.repo_exists(input_model_id):
-            with gr.spinner("Converting model..."):
+            with gr.Progress() as progress:
+                progress(0, desc="Converting model...")
                 success, stderr = converter.convert_model(input_model_id)
                 if not success:
                     return f"Conversion failed: {stderr}"
 
-                gr.Info("Conversion successful!")
-                gr.Code(stderr)
-
-            with gr.spinner("Uploading model..."):
+                progress(0.5, desc="Uploading model...")
                 error = converter.upload_model(input_model_id)
                 if error:
                     return f"Upload failed: {error}"
 
-                gr.Info("Upload successful!")
                 return f"Model uploaded to {input_model_id}/onnx"
         else:
             return f"Model {input_model_id} does not exist on Hugging Face."