Update app.py

app.py

@@ -35,7 +35,10 @@ import altair as alt
 from PIL import Image
 API_URL0 = "https://api-inference.huggingface.co/models/csebuetnlp/mT5_multilingual_XLSum"
 headers0 = {"Authorization": "Bearer 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"}
-
+API_URL1 = "https://api-inference.huggingface.co/models/Michael-Vptn/text-summarization-t5-base"
+headers1 = {"Authorization": "Bearer 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"}
+API_URL2 = "https://api-inference.huggingface.co/models/gpt2"
+headers2 = {"Authorization": "Bearer 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"}
 	
 def read_pdf(file):
 #     images=pdf2image.convert_from_path(file)
@@ -64,9 +67,6 @@ def read_pdf(file):
 #     #st.text(extracted_text)
 #     return extracted_text
 def engsum(output):
-    API_URL1 = "https://api-inference.huggingface.co/models/Michael-Vptn/text-summarization-t5-base"
-    headers1 = {"Authorization": "Bearer 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"}
-    
     def query(payload):
         response = requests.post(API_URL1, headers=headers1, json=payload)
         return response.json()
@@ -155,21 +155,19 @@ def main():
             text = message
 
         if st.checkbox("English Text Generation"): 
-            API_URL2 = "https://api-inference.huggingface.co/models/gpt2"
-            headers2 = {"Authorization": "Bearer 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"}
             
             def query(payload):
             	response = requests.post(API_URL2, headers=headers2, json=payload)
             	return response.json()
             	
-            output = query({
+            out = query({
             	"inputs": text,
             })
-            if isinstance(output, list) and output[0].get("generated_text"):
-                text_output = output[0]["generated_text"]
+            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_ouput)
+                engsum(text_output)
                 
         if st.checkbox("Mark for Text Summarization"):
             if b==1: