Update app.py

app.py

@@ -71,16 +71,20 @@ def engsum(output):
         response = requests.post(API_URL1, headers=headers1, json=payload)
         return response.json()
         
-    output = query({
+    out = query({
         "inputs": output,
     })
-    st.success(output)
+    if isinstance(out, list) and out[0].get("generated_text"):
+        text_output = out[0]["generated_text"]
+    st.success(text_output)
 def bansum(text):
     def query(payload):
         response = requests.post(API_URL0, headers=headers0, json=payload)
         return response.json()
-    output = query({"inputs": text})
-    st.success(output) 
+    out = query({"inputs": text})
+    if isinstance(out, list) and out[0].get("generated_text"):
+        text_output = out[0]["generated_text"]
+    st.success(text_output) 
 
 st.title("NLP APPLICATION")
 #@st.cache_resource(experimental_allow_widgets=True)
@@ -166,8 +170,6 @@ def main():
             if isinstance(out, list) and out[0].get("generated_text"):
                 text_output = out[0]["generated_text"]
                 st.success(text_output)
-            if st.checkbox("Summarize generated text"):
-                engsum(text_output)
                 
         if st.checkbox("Mark for Text Summarization"):
             if b==1: