Update app.py

app.py

@@ -58,7 +58,7 @@ def regenerate_pubkey(pubkey, oauth_token: gr.OAuthToken | None):
         print("Ollama pubkey Generated!")
 
 
-def ollamafy_model(login, model_id, ollama_library_username , ollama_q_method, latest, maintainer, oauth_token: gr.OAuthToken | None):
+def ollamafy_model(login, model_id, ollama_library_username , ollama_q_method, latest, maintainer, oauth_token: gr.OAuthToken | None, ollama_model_name):
     ollama_library_username: library_username | None 
     
     if oauth_token.token is None:
@@ -106,7 +106,8 @@ def ollamafy_model(login, model_id, ollama_library_username , ollama_q_method, l
 
         ### Ollamafy ###
         model_maintainer = model_id.split('/')[-2]
-        ollama_model_name = model_maintainer.lower() + '_' + model_name.lower() 
+        if ollama_model_name = None
+            ollama_model_name = model_maintainer.lower() + '_' + model_name.lower() 
         ollama_modelfile_name = model_name + '_modelfile'
         # model_path = f"{HOME}/.cache/huggingface/hub/{model_id}"
       
@@ -206,16 +207,16 @@ css="""/* Custom CSS to allow scrolling */
 """
 # Create Gradio interface
 with gr.Blocks(css=css) as demo:
-    with gr.Row(): 
-        with gr.Column(scale=1):
-            login = gr.LoginButton(
-                min_width=250,
-            )
-        with gr.Column(scale=1):
-            generate_pubkey = gr.Button (
-                value="Generate Pubkey",
-                min_width=250,
-            )
+    
+    login = gr.LoginButton(
+        min_width=250,
+    )
+
+    generate_pubkey = gr.Button (
+        value="Generate Pubkey",
+        min_width=250,
+    )
+    
     model_id = HuggingfaceHubSearch(
         label="Hugging Face Hub Model ID",
         placeholder="Search for model id on Huggingface",
@@ -234,29 +235,36 @@ with gr.Blocks(css=css) as demo:
         label="Copy this and paste this into your Ollama profile.",
     )
     
+    ollama_library_username = gr.Textbox(
+        label="Ollama Library Username",
+        info="Input your username from Ollama to Push this model to their Library.",
+    )
+
     ollama_library_username = gr.Textbox(
         label="Ollama Library Username",
         info="Input your username from Ollama to Push this model to their Library.",
     )
     
-    with gr.Row(): 
-        with gr.Column(scale=1):
-            
-            latest = gr.Checkbox(
-                value=False,
-                label="Latest",
-                info="Push Model to the Ollama Library with the :latest tag."
-            )
+    latest = gr.Checkbox(
+        value=False,
+        label="Latest",
+        info="Push Model to the Ollama Library with the :latest tag."
+    )
 
-        with gr.Column(scale=1):
-            maintainer = gr.Checkbox(
-                value=False,
-                label="Maintainer",
-                info="Use this option is your original repository on both Hugging Face and Ollama."
-            )
+    maintainer = gr.Checkbox(
+        value=False,
+        label="Maintainer",
+        info="Use this option is your original repository on both Hugging Face and Ollama."
+    )
             
     generate_pubkey.click(
         fn=regenerate_pubkey,
+        inputs=[
+            generate_pubkey
+        ],
+        outputs=[
+            pubkey
+        ],
     )
     
     iface = gr.Interface(
@@ -265,6 +273,7 @@ with gr.Blocks(css=css) as demo:
             login,
             generate_pubkey,
             model_id,
+            ollama_model_name,
             ollama_library_username,
             ollama_q_method,
             latest,