Update app.py

app.py

@@ -67,14 +67,12 @@ def regenerate_pubkey(pubkey, oauth_token: gr.OAuthToken | None):
             raise Exception(f"Error generating Ollama Host File {result.stderr}")
         print("Ollama Host File Created!")
 
-        
     except Exception as e:
         return (f"Error: {e}", "error.png")
     finally:
         # shutil.rmtree(model_name, ignore_errors=True)
         print("Ollama Pubkey Generated! Copy to your user profile in the Ollama Library.")
 
-
 def ollamafy_model(login, model_id, ollama_library_username , ollama_q_method, latest, download_gguf_link, maintainer, oauth_token: gr.OAuthToken | None, ollama_model_name):
     ollama_library_username: library_username | None 
     
@@ -178,7 +176,6 @@ def ollamafy_model(login, model_id, ollama_library_username , ollama_q_method, l
             raise Exception(f"Error removing to Ollama: {ollama_rm_result.stderr}")
         else: 
             print("Model pushed to Ollama library successfully!")
-      
 
         if latest:
             ollama_copy =  f"ollama cp {library_username}/{model_id.lower()}:{q_method.lower()} {library_username}/{model_id.lower()}:latest"
@@ -204,12 +201,11 @@ def ollamafy_model(login, model_id, ollama_library_username , ollama_q_method, l
 
             ollama_rm_latest_result = subprocess.run(ollama_rm_latest, shell=True, capture_output=True)
             print(ollama_rm_latest_result)
-            if ollama_rm_latest_result.returncode != 0:
-                raise Exception(f"Error pushing to Ollama: {ollama_rm_latest.stderr}")
-            else:
-                print("Model pushed to Ollama library successfully!")
+        if ollama_rm_latest_result.returncode != 0:
+            raise Exception(f"Error pushing to Ollama: {ollama_rm_latest.stderr}")
+        else:
+            print("Model pushed to Ollama library successfully!")
     
-
     except Exception as e:
         return (f"Error: {e}", "error.png")
     finally:
@@ -227,7 +223,6 @@ def ollamafy_model(login, model_id, ollama_library_username , ollama_q_method, l
         if result.returncode != 0:
             raise Exception(f"Error removing Ollama HOME folder {result.stderr}")
         print("Ollama HOME fodler removed successfully!")
-
         
 css="""/* Custom CSS to allow scrolling */
 .gradio-container {overflow-y: auto;}