File size: 827 Bytes
db57927 |
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 |
import shutil
import gradio as gr
from fastapi import FastAPI
from fastapi import status
from fastapi.responses import JSONResponse
from modules.api.models import *
from modules.api import api
GRADIO_CACHE_DIR = '/tmp/gradio'
def sd_gradio_cleaner_api(_: gr.Blocks, app: FastAPI):
@app.post("/clean_cache")
async def clean_cache():
try:
shutil.rmtree(GRADIO_CACHE_DIR)
except Exception as e:
return JSONResponse(
content=str(e),
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
)
return JSONResponse(
content="Ok",
status_code=status.HTTP_200_OK,
)
try:
import modules.script_callbacks as script_callbacks
script_callbacks.on_app_started(sd_gradio_cleaner_api)
except:
pass
|