GOGO198 commited on
Commit
223da69
·
verified ·
1 Parent(s): 2e36b27

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -45
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应用 (替代Gradio Blocks)
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
- threading.Thread(target=run_fastapi, daemon=True).start()
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)