Update main.py

main.py

@@ -157,8 +157,9 @@ def getChat(text,model,tokenizer):
     f = 0
     if tt[0][0].isdigit():
         size_float = [float(i) for i in tt]
-        if size_float>4:
-            f = 1
+        for  kk in size_float:
+            if kk>=4:
+                f = 1
     else:
         size_str = "无"