sh20raj commited on
Commit
c8aa0ff
1 Parent(s): 32686d2

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +30 -0
app.py ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ from flask import Flask, request, redirect, render_template
3
+ from huggingface_hub import HfApi
4
+
5
+ app = Flask(__name__)
6
+ api = HfApi()
7
+
8
+ @app.route('/', methods=['GET', 'POST'])
9
+ def upload_file():
10
+ if request.method == 'POST':
11
+ file = request.files['file']
12
+ repo_id = request.form['repo_id']
13
+ revision = request.form['revision']
14
+
15
+ try:
16
+ api.upload_file(
17
+ path_or_file=file,
18
+ path_in_repo=file.filename,
19
+ repo_id=repo_id,
20
+ revision=revision,
21
+ create_pr=True,
22
+ )
23
+ return f'File {file.filename} uploaded successfully to {repo_id}'
24
+ except Exception as e:
25
+ return str(e)
26
+
27
+ return render_template('index.html')
28
+
29
+ if __name__ == '__main__':
30
+ app.run(debug=True)