trainingdata / app.py
sh20raj's picture
Create app.py
c8aa0ff verified
raw
history blame
No virus
822 Bytes
import os
from flask import Flask, request, redirect, render_template
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']
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)
return render_template('index.html')
if __name__ == '__main__':
app.run(debug=True)