Update main.py
Browse files
main.py
CHANGED
@@ -142,7 +142,32 @@ def getChat(text,model,tokenizer):
|
|
142 |
size_list.append((start,end))
|
143 |
start = end
|
144 |
# print(size_list)
|
145 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
146 |
# if len(size_list)==0:
|
147 |
# size_str = "无"
|
148 |
# else:
|
@@ -161,7 +186,7 @@ def getChat(text,model,tokenizer):
|
|
161 |
# print(cancer,transfer,ly_transfer)
|
162 |
output = "肿瘤大小:"+size_str+"\n肿瘤大小<>=4cm:"+"无"+"\n"+"病理组织分型:"+cancer+"\n"+"转移:"+transfer+"\n"+"淋巴转移:"+ly_transfer+"\n"
|
163 |
|
164 |
-
return output
|
165 |
|
166 |
app = FastAPI()
|
167 |
|
@@ -183,9 +208,9 @@ tokenizer,model_extra = model_init()
|
|
183 |
|
184 |
@app.get("/infer_t5")
|
185 |
def t5(input):
|
186 |
-
output = getChat(input,model_extra,tokenizer)
|
187 |
# output = pipe_flan(input)
|
188 |
-
return {"output": output}
|
189 |
|
190 |
app.mount("/", StaticFiles(directory="static", html=True), name="static")
|
191 |
|
|
|
142 |
size_list.append((start,end))
|
143 |
start = end
|
144 |
# print(size_list)
|
145 |
+
size_text = []
|
146 |
+
for k in size_list:
|
147 |
+
size_text.append(text[k[0]:k[1]])
|
148 |
+
if len(size_text)==0:
|
149 |
+
size_str = "无"
|
150 |
+
else:
|
151 |
+
size_str = size_text[0]
|
152 |
+
if '×' in s:
|
153 |
+
split_w = '×'
|
154 |
+
else:
|
155 |
+
split_w = '*'
|
156 |
+
tt = s.split(split_w)
|
157 |
+
f = 0
|
158 |
+
if tt[0][0].isdigit():
|
159 |
+
size_float = [float(i) for i in tt]
|
160 |
+
if size_float>4:
|
161 |
+
f = 1
|
162 |
+
else:
|
163 |
+
size_str = "无"
|
164 |
+
|
165 |
+
if size_str == "无":
|
166 |
+
size_4 = "无"
|
167 |
+
elif f==0:
|
168 |
+
size_4 = "<4cm"
|
169 |
+
else:
|
170 |
+
size_4 = ">=4cm"
|
171 |
# if len(size_list)==0:
|
172 |
# size_str = "无"
|
173 |
# else:
|
|
|
186 |
# print(cancer,transfer,ly_transfer)
|
187 |
output = "肿瘤大小:"+size_str+"\n肿瘤大小<>=4cm:"+"无"+"\n"+"病理组织分型:"+cancer+"\n"+"转移:"+transfer+"\n"+"淋巴转移:"+ly_transfer+"\n"
|
188 |
|
189 |
+
return output,size_str,size_4,cancer,transfer,ly_transfer
|
190 |
|
191 |
app = FastAPI()
|
192 |
|
|
|
208 |
|
209 |
@app.get("/infer_t5")
|
210 |
def t5(input):
|
211 |
+
output,size_str,size_4,cancer,transfer,ly_transfer = getChat(input,model_extra,tokenizer)
|
212 |
# output = pipe_flan(input)
|
213 |
+
return {"output": output,"size_str":size_str,"size_4":size_4,"cancer":cancer,"transfer":transfer,"ly_transfer":ly_transfer}
|
214 |
|
215 |
app.mount("/", StaticFiles(directory="static", html=True), name="static")
|
216 |
|