Update main.py

main.py

@@ -1,20 +1,36 @@
+# 从 FastAPI 库中导入 FastAPI 类，用于创建应用实例
 from fastapi import FastAPI
+
+# 从 FastAPI 的 staticfiles 模块中导入 StaticFiles 类，用于处理静态文件
 from fastapi.staticfiles import StaticFiles
+
+# 从 FastAPI 的 responses 模块中导入 FileResponse 类，用于返回文件响应
 from fastapi.responses import FileResponse
 
+# 从 transformers 库中导入 pipeline 函数，用于加载预训练模型管道
 from transformers import pipeline
 
+# 创建 FastAPI 应用实例
 app = FastAPI()
 
+# 使用 transformers 的 pipeline 函数加载 Flan-T5 小模型，用于文本生成任务
 pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
 
+# 定义一个 GET 请求的路由，路径为 /infer_t5，用于处理文本生成请求
 @app.get("/infer_t5")
+# 定义处理 /infer_t5 路由的函数，接收一个输入参数
 def t5(input):
+    # 使用加载的 Flan-T5 模型管道生成输出
     output = pipe_flan(input)
+    # 返回生成的文本结果，格式为 JSON
     return {"output": output[0]["generated_text"]}
 
+# 挂载静态文件目录，将根路径 "/" 指向 "static" 目录，并启用 HTML 模式
 app.mount("/", StaticFiles(directory="static", html=True), name="static")
 
+# 定义一个 GET 请求的路由，路径为根路径 "/"，用于返回首页
 @app.get("/")
+# 定义处理根路径的函数，返回一个文件响应，类型为 HTML
 def index() -> FileResponse:
+    # 返回位于 /app/static/index.html 的 HTML 文件
     return FileResponse(path="/app/static/index.html", media_type="text/html")