Update app.py

app.py

@@ -32,7 +32,7 @@ def fetch_manifest(image, tag):
 
 def upload_to_huggingface(repo_id, folder_path):
     api = HfApi(token=HF_TOKEN)
-    repo_path = api.create_repo(repo_id, "model", exist_ok=True,)
+    repo_path = api.create_repo(repo_id, "model", exist_ok=True)
     print(f"Repo created {repo_path}")
     try:
         api.upload_folder(
@@ -73,28 +73,28 @@ def process_image_tag(image_tag, repo_id):
             download_file(config_digest, image)
     
         # Upload to Hugging Face Hub
-        upload_result = upload_to_huggingface(repo_id, 'blobs/*')
+        upload_result = upload_to_huggingface(repo_id, 'blobs')
     
         # Delete the blobs folder
-    
         os.rmtree('blobs')
         return f"Successfully fetched and downloaded files for {image}:{tag}\n{upload_result}\nBlobs folder deleted"
     except Exception as e:
-        os.rmtree('blobs')
+        os.rmtree('blobs', ignore_errors=True)
         return f"Error found: {str(e)}"
 
-# Create the Gradio interface
-iface = gr.Interface(
-    fn=process_image_tag,
-    inputs=[
-        gr.Textbox(placeholder="Enter Ollama ID", label="Image and Tag"),
-        gr.Textbox(placeholder="Enter Hugging Face repo ID", label="Hugging Face Repo ID"),
-    )
-    ],
-    outputs=gr.Textbox(label="Result"),
-    title="Ollama <> HF Hub 🤝",
-    description="Enter the image and tag to download the corresponding files from the Ollama registry and upload them to the Hugging Face Hub."
-)
+# Create the Gradio interface using gr.Blocks
+with gr.Blocks() as demo:
+    gr.Markdown("# Ollama <> HF Hub 🤝")
+    gr.Markdown("Enter the image and tag to download the corresponding files from the Ollama registry and upload them to the Hugging Face Hub.")
+    
+    with gr.Row():
+        image_tag_input = gr.Textbox(placeholder="Enter Ollama ID", label="Image and Tag")
+        repo_id_input = gr.Textbox(placeholder="Enter Hugging Face repo ID", label="Hugging Face Repo ID")
+    
+    result_output = gr.Textbox(label="Result")
+    
+    process_button = gr.Button("Process")
+    process_button.click(fn=process_image_tag, inputs=[image_tag_input, repo_id_input], outputs=result_output)
 
 # Launch the Gradio app
-iface.launch()
+demo.launch()