Update app.py

app.py

@@ -18,10 +18,10 @@ MODELS = {
         "sizes": {
             "1.5B": "Spestly/Atlas-R1-1.5B-Preview",
         },
-        "emoji": "",
+        "emoji": "🦁",
         "experimental": True,
-        "is_vision": False,  # Set to True to enable vision support
-        "system_prompt_env": "ATLAS_FLASH_1215",  # Env variable for the system prompt
+        "is_vision": False,  # Enable vision support for this model
+        "system_prompt_env": "ATLAS_FLASH_1215",  # Environment variable for system prompt
     },
 }
 
@@ -176,7 +176,7 @@ class AtlasInferenceApp:
             ):
                 st.markdown(message["content"])
                 if "image" in message:
-                    st.image(message["image"], caption="Uploaded Image", use_column_width=True)
+                    st.image(message["image"], caption="Uploaded Image", use_container_width=True)  # Updated parameter
 
         # Input box for user messages
         if prompt := st.chat_input("Message Atlas..."):
@@ -189,7 +189,7 @@ class AtlasInferenceApp:
             with st.chat_message("user", avatar=USER_PFP):
                 st.markdown(prompt)
                 if uploaded_image:
-                    st.image(uploaded_image, caption="Uploaded Image", use_column_width=True)
+                    st.image(uploaded_image, caption="Uploaded Image", use_container_width=True)  # Updated parameter
 
             with st.chat_message("assistant", avatar=AI_PFP):
                 with st.spinner("Generating response..."):