Update app.py

app.py

@@ -146,10 +146,10 @@ def login_section():
 
 
 
-def predict(cap_col, target_language):
+def predict(cap_col, img, target_language):
     captions = []
-    url = 'tmp.jgp'
-    pred_caption = image_to_text(url)
+    
+    pred_caption = image_to_text(img)
 
     cap_col.markdown('#### Predicted Captions:')
     translated_caption = translate(pred_caption, target_language)
@@ -173,8 +173,7 @@ def generate_caption_section():
     # Image URL input
     img_url = st.text_input(label='Enter Image URL')
 
-    # Image upload input
-    img_upload = st.file_uploader(label='Upload Image', type=['jpg', 'png', 'jpeg'])
+    
 
     # Language selection dropdown
            
@@ -187,20 +186,14 @@ def generate_caption_section():
         img = img.convert('RGB')
         col1.image(img, caption="Input Image", use_column_width=True)
         img.save('tmp.jpg')
-        predict(col2, target_language)
+        predict(col2, img, target_language)
 
         st.markdown('<center style="opacity: 70%">OR</center>', unsafe_allow_html=True)
 
-    elif img_upload:
-        img = img_upload.read()
-        img = Image.open(io.BytesIO(img))
-        img = img.convert('RGB')
-        col1.image(img, caption="Input Image", use_column_width=True)
-        img.save('tmp.jpg')
-        predict(col2, target_language)
+
 
     # Remove temporary image file
-    if img_url or img_upload:
+    if img_url
         os.remove('tmp.jpg')
 
 def main():