99i commited on
Commit
934814f
·
verified ·
1 Parent(s): 4445920

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -2
main.py CHANGED
@@ -9,7 +9,8 @@ from fastapi.staticfiles import StaticFiles
9
 
10
  app = FastAPI()
11
  # 挂载 tmp 文件夹下的静态文件
12
- app.mount("/static", StaticFiles(directory="tmp"), name="static")
 
13
  # 添加跨域支持
14
  origins = [
15
  "*"
@@ -37,7 +38,7 @@ async def generate_image_endpoint(markdown: Markdown):
37
  # 调用 generate_image 函数来生成图片
38
  generate_image(markdown.markdown, image_filename)
39
  # 返回图片的路径
40
- return {"image_path": f"{image_filename}"}
41
 
42
  # 创建一个函数,用于清理过期的图片
43
  def clean_expired_images():
 
9
 
10
  app = FastAPI()
11
  # 挂载 tmp 文件夹下的静态文件
12
+ app.mount("/tmp", StaticFiles(directory="tmp"), name="static")
13
+ host=os.environ.get("HOST", "https://99i-t2c.hf.space")
14
  # 添加跨域支持
15
  origins = [
16
  "*"
 
38
  # 调用 generate_image 函数来生成图片
39
  generate_image(markdown.markdown, image_filename)
40
  # 返回图片的路径
41
+ return {"url": f"{host}/{image_filename}"}
42
 
43
  # 创建一个函数,用于清理过期的图片
44
  def clean_expired_images():