Wind-xixi commited on
Commit
5ceb402
·
verified ·
1 Parent(s): cd1cf6e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -2,7 +2,7 @@ from fastapi import FastAPI, File, UploadFile
2
  from fastapi.responses import JSONResponse
3
  import uvicorn
4
  import os
5
- from predictor import SentenceExtractor # 改成直接导入你修改后的类
6
 
7
  # 创建 FastAPI 应用
8
  app = FastAPI()
@@ -35,8 +35,6 @@ async def evaluate_file(file: UploadFile = File(...)):
35
  "中性词语评价数": result["neutral_word_count"],
36
  "句子评分": []
37
  }
38
-
39
- # 添加句子评分信息
40
  for i, item in enumerate(result["scored_sentences"], 1):
41
  formatted_result["句子评分"].append({
42
  f"句子{i}": f"{item['sentence']} - {item['grade']}"
@@ -48,7 +46,7 @@ async def evaluate_file(file: UploadFile = File(...)):
48
  return JSONResponse(content={"error": str(e)}, status_code=500)
49
 
50
  if __name__ == "__main__":
51
- # Hugging Face Spaces 会设置环境变量 HF_SPACE
52
- port = int(os.getenv("PORT", 7860)) # HF Spaces 会传 PORT
53
- uvicorn.run("app:app", host="0.0.0.0", port=port)
54
 
 
2
  from fastapi.responses import JSONResponse
3
  import uvicorn
4
  import os
5
+ from predictor import SentenceExtractor # 保证导入规范文件名 `predictor.py`
6
 
7
  # 创建 FastAPI 应用
8
  app = FastAPI()
 
35
  "中性词语评价数": result["neutral_word_count"],
36
  "句子评分": []
37
  }
 
 
38
  for i, item in enumerate(result["scored_sentences"], 1):
39
  formatted_result["句子评分"].append({
40
  f"句子{i}": f"{item['sentence']} - {item['grade']}"
 
46
  return JSONResponse(content={"error": str(e)}, status_code=500)
47
 
48
  if __name__ == "__main__":
49
+ port = int(os.getenv("PORT", 7860))
50
+ uvicorn.run(app, host="0.0.0.0", port=port)
51
+
52