Spaces:
Runtime error
Runtime error
File size: 1,525 Bytes
d760545 a18602d c2f31df 0d68e6c c2f31df d760545 c2f31df 0cb9be7 a18602d c2f31df 3b04c28 0cb9be7 0d68e6c 3b04c28 0cb9be7 3b04c28 654fde1 2636cc8 654fde1 0cb9be7 39c1111 d760545 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
from huggingface_hub import HfApi
from huggingface_hub import hf_hub_download
import huggingface_hub
from huggingface_hub import get_repo_discussions
from bs4 import BeautifulSoup
import json
import re
from fastapi import FastAPI
from fastapi import Request
repo_id = "dalle-mini/dalle-mini"
repo_id = "stabilityai/stable-diffusion"
dataset_repo_id = "triple-t/dummy"
path = "image_cache"
file_name = "_".join(repo_id.split("/")) + ".json"
api = HfApi()
app = FastAPI()
items = []
url_re = re.compile(r'https://s3\.amazonaws\.com/moonup/production/uploads/\d+-noauth\.jpeg')
@app.get("/")
def get_events():
return items
@app.post("/webhook")
def webhook(request: Request):
global items
payload = request.json()
event = payload["event"]
if event["action"] != "create":
return
if event["scope"] != "discussion":
return
content = payload["comment"]["content"]
title = payload["discussion"]["title"]
image_urls = url_re.findall(content)
data_dict = {"data": {"images": image_urls, "prompt": title}, "discussion_num": 1000}
path = hf_hub_download(repo_id=dataset_repo_id, filename=file_name, cache_dir=path, repo_type="dataset")
with open(path, "r") as f:
data = json.load(f)
data.append(data_dict)
with open(path, "w") as f:
f.write(json.dumps(data, sort_keys=True, indent=4))
api.upload_file(
path_or_fileobj=path,
path_in_repo=file_name,
repo_id=dataset_repo_id,
repo_type="dataset",
)
|