fix

app.py

@@ -88,30 +88,31 @@ def neuron_export(model_id: str, task: str, oauth_token: gr.OAuthToken) -> str:
     try:
         api = HfApi(token=oauth_token.token)
 
-        error, result_info = convert(api=api, model_id=model_id, task=task, token=oauth_token.token)
+        error, commit_info = convert(api=api, model_id=model_id, task=task, token=oauth_token.token)
         if error != "0":
             return error
 
-        print("[result_info]", result_info)
+        print("[commit_info]", commit_info)
 
-        # Save in a private dataset if repo initialized
+         # Save in a private dataset if repo initialized
         if repo is not None:
             repo.git_pull(rebase=True)
             with open(os.path.join(DATADIR, DATA_FILE), "a") as csvfile:
                 writer = csv.DictWriter(
-                    csvfile, fieldnames=["model_id", "cache_info", "time"]
+                    csvfile, fieldnames=["model_id", "pr_url", "time"]
                 )
                 writer.writerow(
                     {
                         "model_id": model_id,
-                        "cache_info": str(result_info),
+                        "pr_url": commit_info.pr_url,
                         "time": str(datetime.now()),
                     }
                 )
             commit_url = repo.push_to_hub()
             print("[dataset]", commit_url)
 
-        return f"#### Success 🔥 Yay! This model was successfully compiled and cached! The compilation artifacts have been synchronized to the optimum-neuron cache repository. You can now use this model with optimum-neuron and benefit from faster loading times. Model: {model_id}, Task: {task}"
+        pr_revision = commit_info.pr_revision.replace("/", "%2F")
+        return f"#### Success 🔥 Yay! This model was successfully exported and a PR was opened using your token: [{commit_info.pr_url}]({commit_info.pr_url}). If you would like to use the exported model without waiting for the PR to be approved, head to https://huggingface.co/{model_id}/tree/{pr_revision}"
 
     except Exception as e:
         return f"#### Error: {e}"