Update main.py

main.py

@@ -103,7 +103,7 @@ class Word_BERT(nn.Module):
         ly_transfer = self.ly_transfer(pool)
         return out,cancer,transfer,ly_transfer
 
-def getChat(text):
+def getChat(text,model):
     # while True:
     #     if True:
     # text = input("user:")
@@ -141,7 +141,7 @@ def getChat(text):
             end = i
             size_list.append((start,end))
             start = end
-    print(size_list)
+    # print(size_list)
 
     cancer_dict = {'腺癌': 0, '肺良性疾病': 1, '鳞癌': 2, '无法判断组织分型': 3, '复合型': 4, '转移癌': 5, '小细胞癌': 6, '大细胞癌': 7}
     id_cancer = {j:i for i,j in cancer_dict.items()}
@@ -175,13 +175,13 @@ def model_init():
     return model
     # return None
 
-model11 = model_init()
-print(os.getcwd())
+model_extra = model_init()
 
 @app.get("/infer_t5")
 def t5(input):
-    output = pipe_flan(input)
-    return {"output": output[0]["generated_text"]}
+    r1,r2,r3,r4 = getChat(input,model_extra)
+    # output = pipe_flan(input)
+    return {"output": r4}
 
 app.mount("/", StaticFiles(directory="static", html=True), name="static")