Update app.py

app.py

@@ -59,19 +59,20 @@ def engsum(output):
     out = query({
         "inputs": output,
     })
-    #if isinstance(out, list) and out[0].get("generated_text"):
-        #text_output = out[0]["generated_text"]
-    st.success(out)
+    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()
     out = query({"inputs": text})
-    #if isinstance(out, list) and out[0].get("generated_text"):
-        #text_output = out[0]["generated_text"]
-    st.success(out) 
+    if isinstance(out, list) and out[0].get("summary_text"):
+       text_output = out[0]["summary_text"]
+       st.success(text_output) 
 
-st.title("NLP APPLICATION")
+st.title("Bangla and English Summarizer:")
+st.subheader("Please, Upload your PDF/Images or write texts to see the summarize options: ")
 #@st.cache_resource(experimental_allow_widgets=True)
 def main():
     #global tokenizer, model
@@ -85,7 +86,6 @@ def main():
     	""")
     def change_photo_state():
         st.session_state["photo"]="done"
-    st.subheader("Please, feed your pdf/images/text, features/services will appear automatically!")
     message = st.text_input("Type your text here!")
     uploaded_photo = st.sidebar.file_uploader("Upload your Images/PDF",type=['jpg','png','jpeg','pdf'], on_change=change_photo_state)
     camera_photo = st.sidebar.camera_input("Take a photo, Containing English texts", on_change=change_photo_state)
@@ -128,11 +128,11 @@ def main():
                 engsum(text)
         elif uploaded_photo==None and camera_photo==None:
             text = message
-            if st.button("Content Type: Bangla"):
+            if st.button("Bangla"):
                 bansum(text)
-            if st.button("Content Type: English"): 
+            if st.button("English"): 
                 engsum(text)  
-        if st.checkbox("English Text Generation"): 
+        if st.button("English Text Generation"): 
             def query(payload):
             	response = requests.post(API_URL2, headers=headers2, json=payload)
             	return response.json()