Update app.py
Browse files
app.py
CHANGED
@@ -6,6 +6,12 @@ import faiss
|
|
6 |
from huggingface_hub import hf_hub_download
|
7 |
import time
|
8 |
import json
|
|
|
|
|
|
|
|
|
|
|
|
|
9 |
|
10 |
# 创建安全缓存目录
|
11 |
CACHE_DIR = "/home/user/cache"
|
@@ -131,12 +137,7 @@ def predict(vector):
|
|
131 |
"message": "处理错误,请重试或联系管理员"
|
132 |
}
|
133 |
|
134 |
-
# 创建FastAPI应用
|
135 |
-
import fastapi
|
136 |
-
from fastapi import FastAPI, Request
|
137 |
-
from fastapi.responses import JSONResponse
|
138 |
-
from fastapi.middleware.cors import CORSMiddleware
|
139 |
-
|
140 |
app = FastAPI()
|
141 |
|
142 |
# 添加CORS支持
|
@@ -173,44 +174,8 @@ async def api_predict(request: Request):
|
|
173 |
}
|
174 |
)
|
175 |
|
176 |
-
# 保留Gradio界面
|
177 |
-
with gr.Blocks() as demo:
|
178 |
-
gr.Markdown("## 🛍 电商智能客服系统 (768维专用)")
|
179 |
-
gr.Markdown("**使用CLIP-vit-large-patch14模型 | 向量维度: 768**")
|
180 |
-
|
181 |
-
with gr.Row():
|
182 |
-
vector_input = gr.Dataframe(
|
183 |
-
headers=["向量值"],
|
184 |
-
type="array",
|
185 |
-
label="输入向量 (768维)",
|
186 |
-
value=[[0.1]*768]
|
187 |
-
)
|
188 |
-
output = gr.JSON(label="智能回答")
|
189 |
-
|
190 |
-
submit_btn = gr.Button("生成回答")
|
191 |
-
submit_btn.click(
|
192 |
-
fn=predict,
|
193 |
-
inputs=vector_input,
|
194 |
-
outputs=output
|
195 |
-
)
|
196 |
-
|
197 |
# 启动应用
|
198 |
if __name__ == "__main__":
|
199 |
-
import uvicorn
|
200 |
-
import threading
|
201 |
-
|
202 |
-
# 启动FastAPI服务
|
203 |
-
def run_fastapi():
|
204 |
-
uvicorn.run(app, host="0.0.0.0", port=7860)
|
205 |
-
|
206 |
-
# 启动Gradio界面
|
207 |
-
def run_gradio():
|
208 |
-
demo.launch(
|
209 |
-
server_name="0.0.0.0",
|
210 |
-
server_port=7861,
|
211 |
-
share=False
|
212 |
-
)
|
213 |
-
|
214 |
# 验证资源
|
215 |
print("="*50)
|
216 |
print("Space启动完成 | 准备接收请求")
|
@@ -218,6 +183,5 @@ if __name__ == "__main__":
|
|
218 |
print(f"元数据记录数: {len(metadata)}")
|
219 |
print("="*50)
|
220 |
|
221 |
-
#
|
222 |
-
|
223 |
-
run_gradio()
|
|
|
6 |
from huggingface_hub import hf_hub_download
|
7 |
import time
|
8 |
import json
|
9 |
+
import fastapi
|
10 |
+
from fastapi import FastAPI, Request
|
11 |
+
from fastapi.responses import JSONResponse
|
12 |
+
from fastapi.middleware.cors import CORSMiddleware
|
13 |
+
import uvicorn
|
14 |
+
import threading
|
15 |
|
16 |
# 创建安全缓存目录
|
17 |
CACHE_DIR = "/home/user/cache"
|
|
|
137 |
"message": "处理错误,请重试或联系管理员"
|
138 |
}
|
139 |
|
140 |
+
# 创建FastAPI应用
|
|
|
|
|
|
|
|
|
|
|
141 |
app = FastAPI()
|
142 |
|
143 |
# 添加CORS支持
|
|
|
174 |
}
|
175 |
)
|
176 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
177 |
# 启动应用
|
178 |
if __name__ == "__main__":
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
179 |
# 验证资源
|
180 |
print("="*50)
|
181 |
print("Space启动完成 | 准备接收请求")
|
|
|
183 |
print(f"元数据记录数: {len(metadata)}")
|
184 |
print("="*50)
|
185 |
|
186 |
+
# 只启动FastAPI服务
|
187 |
+
uvicorn.run(app, host="0.0.0.0", port=7860)
|
|