Update app.py
Browse files
app.py
CHANGED
|
@@ -5,7 +5,7 @@ import mimetypes
|
|
| 5 |
from datetime import datetime
|
| 6 |
from io import BytesIO
|
| 7 |
|
| 8 |
-
from flask import Flask, request, jsonify, redirect
|
| 9 |
from huggingface_hub import HfApi, HfFileSystem
|
| 10 |
from PIL import Image
|
| 11 |
import uuid
|
|
@@ -40,9 +40,9 @@ def save_to_dataset(content, filename):
|
|
| 40 |
fs.write_bytes(f"datasets/{DATASET_ID}/images/{filename}", content)
|
| 41 |
return f"https://huggingface.co/datasets/{DATASET_ID}/resolve/main/images/{filename}"
|
| 42 |
|
| 43 |
-
@app.route("/
|
| 44 |
def index():
|
| 45 |
-
return "
|
| 46 |
|
| 47 |
@app.route("/api/1/upload", methods=["GET", "POST"])
|
| 48 |
def upload():
|
|
@@ -122,4 +122,5 @@ def upload():
|
|
| 122 |
})
|
| 123 |
|
| 124 |
if __name__ == "__main__":
|
| 125 |
-
app.run(host="0.0.0.0", port=7860)
|
|
|
|
|
|
| 5 |
from datetime import datetime
|
| 6 |
from io import BytesIO
|
| 7 |
|
| 8 |
+
from flask import Flask, request, jsonify, redirect, render_template
|
| 9 |
from huggingface_hub import HfApi, HfFileSystem
|
| 10 |
from PIL import Image
|
| 11 |
import uuid
|
|
|
|
| 40 |
fs.write_bytes(f"datasets/{DATASET_ID}/images/{filename}", content)
|
| 41 |
return f"https://huggingface.co/datasets/{DATASET_ID}/resolve/main/images/{filename}"
|
| 42 |
|
| 43 |
+
@app.route("/", methods=["GET"])
|
| 44 |
def index():
|
| 45 |
+
return render_template("index.html")
|
| 46 |
|
| 47 |
@app.route("/api/1/upload", methods=["GET", "POST"])
|
| 48 |
def upload():
|
|
|
|
| 122 |
})
|
| 123 |
|
| 124 |
if __name__ == "__main__":
|
| 125 |
+
app.run(host="0.0.0.0", port=7860, debug=True)
|
| 126 |
+
|