Update app.py

app.py

@@ -224,7 +224,7 @@ def display_results(result, model_key):
     
     elif model_key == 'image_segmentation':
         st.subheader("🧩 Segmentação")
-        st.image(result[0]['mask'], caption="Máscara de segmentação", use_container_width=True)
+        st.image(result[0]['mask'], caption="Máscara de segmentação")
     
     elif model_key == 'facial_recognition':
         st.subheader("😊 Reconhecimento Facial")
@@ -242,7 +242,7 @@ def display_results(result, model_key):
     
     elif model_key == 'text_to_image':
         st.subheader("🎨 Imagem Gerada")
-        st.image(result[0], caption="Imagem gerada a partir do texto", use_container_width=True)
+        st.image(result[0], caption="Imagem gerada a partir do texto")
 
 def main():
     st.title("🤖 Aplicação de IA Multi-Modal Avançada")