Update app.py

app.py

@@ -35,12 +35,15 @@ import altair as alt
 from transformers import AutoTokenizer, AutoModelWithLMHead
 #pytesseract.pytesseract.tesseract_cmd = r"./Tesseract-OCR/tesseract.exe"
 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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"}
+headers0 = {"Authorization": "Bearer "+str(os.environ["t5multilingual"])}
+
+# API_URL1 = "https://api-inference.huggingface.co/models/Michael-Vptn/text-summarization-t5-base"
+# headers1 = {"Authorization": "Bearer "+str(os.environ["t5modeleng"])}
+
+# API_URL2 = "https://api-inference.huggingface.co/models/gpt2"
+# headers2 = {"Authorization": "Bearer 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"}
 @st.cache
 def read_pdf(file):
 #     images=pdf2image.convert_from_path(file)
@@ -94,15 +97,15 @@ def main():
     def change_photo_state():
         st.session_state["photo"]="done"
     with st.container():
-        c1, c2, c3 = st.columns([2,2,1])
+        c1, c2, c3 = st.columns([2,1,2])
         message = c1.text_input("Type your text here!")
         Capture=True
-        if c2.button("Start Camera"):
-           camera_photo = c2.camera_input("Capture a photo to summarize: ", on_change=change_photo_state)
-        if c2.button("Stop Camera"):
+        if c3.button("Start Camera"):
+           camera_photo = c3.camera_input("Capture a photo to summarize: ", on_change=change_photo_state)
+        if c3.button("Stop Camera"):
            Capture =False
            
-        uploaded_photo = c3.file_uploader("Upload your Images/PDF",type=['jpg','png','jpeg','pdf'], on_change=change_photo_state)
+        uploaded_photo = c2.file_uploader("Upload your Images/PDF",type=['jpg','png','jpeg','pdf'], on_change=change_photo_state)
         if st.session_state["photo"]=="done" or message:
             if uploaded_photo and uploaded_photo.type=='application/pdf':
                 tet = read_pdf(uploaded_photo)