chb2026 commited on
Commit
bf2487a
·
verified ·
1 Parent(s): f7c31e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
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("/status", methods=["GET"])
44
  def index():
45
- return "HF-PicBed is running..."
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
+