fix

app.py

@@ -106,8 +106,7 @@ def save_model(model, model_name, quantization_type, group_size=128, username=No
             repo_id=repo_name,
             repo_type="model",
         )
-    
-    return f"https://huggingface.co/{repo_name}"
+    return f'<h1>✅ DONE</h1><br/>Find your repo here: <a href="https://huggingface.co/{repo_name}" target="_blank" style="text-decoration:underline">{repo_name}</a>'
 
 def quantize_and_save(profile: gr.OAuthProfile | None, oauth_token: gr.OAuthToken | None, model_name, quantization_type, group_size, quantized_model_name):
     if oauth_token is None : 
@@ -119,12 +118,11 @@ def quantize_and_save(profile: gr.OAuthProfile | None, oauth_token: gr.OAuthToke
         return exists_message
     if quantization_type == "int4_weight_only" : 
         return "int4_weight_only not supported on cpu"
-    # try : 
     if not group_size.isdigit() :
         return "group_size must be a number"
     
     group_size = int(group_size)
-
+    # try:
     quantized_model = quantize_model(model_name, quantization_type, group_size, oauth_token, profile.username)
     return save_model(quantized_model, model_name, quantization_type, group_size, profile.username, oauth_token, quantized_model_name)
     # except Exception as e : 
@@ -157,7 +155,7 @@ with gr.Blocks(theme=gr.themes.Ocean(), css=css) as app:
                     placeholder="Search for model id on Huggingface",
                     search_type="model",
                 )
-            with gr.Row():
+            with gr.Row(style="margin-top: 30px;"):
                 with gr.Column():
                     quantization_type = gr.Dropdown(
                         label="Quantization Type",