Update app.py

app.py

@@ -64,8 +64,8 @@ def predict(query):
         logits = model(tokens.to(device), attention_mask=mask.to(device))[0]
         probs = logits.softmax(dim=-1)
 
-    real, fake = probs.detach().cpu().flatten().numpy().tolist() # Hello-SimpleAI/chatgpt-detector-roberta       
-#     fake, real = probs.detach().cpu().flatten().numpy().tolist() # PirateXX/AI-Content-Detector-V2
+    # real, fake = probs.detach().cpu().flatten().numpy().tolist() # Hello-SimpleAI/chatgpt-detector-roberta       
+    fake, real = probs.detach().cpu().flatten().numpy().tolist() # PirateXX/AI-Content-Detector-V2
 
     return real
 
@@ -84,9 +84,6 @@ def findRealProb(text):
     realProb = ans/cnt
     return {"Real": realProb, "Fake": 1-realProb}, results
 
-TXT_TO_INSPECT=None
-def inspect_content(text):
-    TXT_TO_INSPECT=text
 
 st.markdown(""" <style> .appview-container .main .block-container {
     max-width: 100%;
@@ -122,7 +119,15 @@ if choose == "Inspect Content":
     with col2:               # To display brand logo                
         st.image('./inspection-1.jpg', width=100 )
 
-    txt = st.text_area('Add Text here',height=300, max_chars=2000, value= '''
+# txt = st.text_area('Text to analyze', '''
+#     It was the best of times, it was the worst of times, it was
+#     the age of wisdom, it was the age of foolishness, it was
+#     the epoch of belief, it was the epoch of incredulity, it
+#     was the season of Light, it was the season of Darkness, it
+#     was the spring of hope, it was the winter of despair, (...)
+#     ''')
+
+    txt = st.text_area('Add Text here','''
     Cristiano Ronaldo is a Portuguese professional soccer player who currently plays 
     as a forward for Manchester United and the Portugal national team. He is widely 
     considered one of the greatest soccer players of all time, having won numerous 
@@ -136,9 +141,10 @@ if choose == "Inspect Content":
     three seasons before returning to Manchester United in 2021. He has also had 
     a successful international career with the Portugal national team, having won 
     the UEFA European Championship in 2016 and the UEFA Nations League in 2019.
-    ''', on_change=inspect_content)
+    ''', height=300, max_chars=2000)
+
 
-    if TXT_TO_INSPECT is not None:
+    if st.button('Analyze Content'): # st.session_state.input_text is not None
         with st.spinner('Loading the model..'):
             model.to(device)
 
@@ -148,9 +154,10 @@ if choose == "Inspect Content":
             # print(audio.shape)
             predictions=findRealProb(txt)
             print('prediction_value',predictions)
-            if predictions[0]['Fake'] > 0.5:
+            if predictions[0]['Fake'] > 0.90:
                 # st.error(f"The Sample is spoof: \n Confidence {(prediction_value) }%",  icon="🚨")
                 st.error(f"This text is AI generated",  icon="🚨")
+
             else:
                 st.success(f"This text is real", icon="✅")
         
@@ -214,7 +221,7 @@ if choose == "Contact":
         Email=st.text_input(label='Please Enter Your Email') #Collect user feedback
         Message=st.text_input(label='Please Enter Your Message') #Collect user feedback
         submitted = st.form_submit_button('Submit')
-        if submitted:
-            st.write('Thanks for your contacting us. We will respond to your questions or inquiries as soon as possible!')
+        # if submitted:
+        #     st.write('Thanks for your contacting us. We will respond to your questions or inquiries as soon as possible!')