Update app.py

app.py

@@ -1,30 +1,31 @@
-import os
-from flask import Flask, request, redirect, render_template
+import gradio as gr
 from huggingface_hub import HfApi
 
-app = Flask(__name__)
 api = HfApi()
 
-@app.route('/', methods=['GET', 'POST'])
-def upload_file():
-    if request.method == 'POST':
-        file = request.files['file']
-        repo_id = request.form['repo_id']
-        revision = request.form['revision']
+def upload_file(file, repo_id, revision):
+    try:
+        api.upload_file(
+            path_or_file=file,
+            path_in_repo=file.name,
+            repo_id=repo_id,
+            revision=revision,
+            create_pr=True,
+        )
+        return f'File {file.name} uploaded successfully to {repo_id}'
+    except Exception as e:
+        return str(e)
 
-        try:
-            api.upload_file(
-                path_or_file=file,
-                path_in_repo=file.filename,
-                repo_id=repo_id,
-                revision=revision,
-                create_pr=True,
-            )
-            return f'File {file.filename} uploaded successfully to {repo_id}'
-        except Exception as e:
-            return str(e)
+file_input = gr.inputs.File(label="File")
+repo_id_input = gr.inputs.Textbox(label="Repository ID")
+revision_input = gr.inputs.Textbox(label="Revision")
 
-    return render_template('index.html')
+iface = gr.Interface(
+    fn=upload_file,
+    inputs=[file_input, repo_id_input, revision_input],
+    outputs="text",
+    title="Upload File to Hugging Face",
+    description="Upload a file to a Hugging Face repository.",
+)
 
-if __name__ == '__main__':
-    app.run(debug=True)
+iface.launch()