Update app.py

app.py

@@ -37,7 +37,8 @@ def humanize(
 ai_check_options = [
     "Polygraf AI",
     "Sapling AI",
-    # "CopyLeaks"
+    # "CopyLeaks",
+    "GPTZero Free"
 ]
 
 def ai_generated_test_polygraf(text):
@@ -81,14 +82,23 @@ def ai_generated_test_copyleak(text):
     print(data.decode("utf-8")["results"]["summary"])
     return data.decode("utf-8")["results"]["summary"]
 
+def ai_generated_test_gptzero(text):
+    from model import GPT2PPL
+    model = GPT2PPL()
+    result = model(text)
+    print(result)
+    return result
+
 
 def ai_check(text, option):
     if option == 'Polygraf AI':
         return ai_generated_test_polygraf(text)
     elif option == 'Sapling AI':
         return ai_generated_test_sapling(text)
-    elif option == 'CopyLeaks':
-        return ai_generated_test_copyleak(text)    
+    # elif option == 'CopyLeaks':
+    #     return ai_generated_test_copyleak(text)    
+    elif option == "GPTZero Free":
+        return ai_generated_test_gptzero(text)
     else:
         return ai_generated_test_polygraf(text)