Update app.py

app.py

@@ -3,8 +3,7 @@
 import simplestart as ss
 import requests
 import json
-import time
-###import mymodule as mm
+import mymodule as mm
 import psutil
 
 
@@ -33,12 +32,19 @@ def predict():
     ss.message(mytext.value)
 
     method = str(ss.session.method)
-    response = inference("'" + mytext.value + "'", method)
-    #response = mm.predict("'" + mytext.value + "'", ss.session.method)
-    #ss.session["result"] = response
-    res = response["processed_text"].replace("<|endoftext|>", "")
-    ss.session["result"] = res
-    ##ss.write(response)
+
+    input = mytext.value[:100].replace("\n", "")
+    
+    #本地模式
+    response = mm.predict("'" + input + "'", ss.session.method)
+    res = response.replace("<|endoftext|>", "")
+
+    #服务器模式
+    #response = inference("'" + input + "'", method)
+    #res = response["processed_text"].replace("<|endoftext|>", "")
+    
+    myresult.value = res
+
 
 def inference(input_text, method):
     #url = "http://0.0.0.0:8106/process"  # 假设POST请求的路径是 /process