Update app.py

app.py

@@ -49,7 +49,7 @@ def generate_caption(image_base64, api_key):
     api_url = "https://api.studio.nebius.ai/v1/chat/completions"
     headers = {"Authorization": f"Bearer {api_key}"}
     payload = {
-        "model": "Qwen/Qwen2-VL-72B-Instruct",
+        "model": "llava-hf/llava-1.5-13b-hf",
         "messages": [
             {
                 "role": "system",
@@ -77,8 +77,7 @@ def generate_caption(image_base64, api_key):
         st.error(f"An exception occurred: {e}")
         return {"error": str(e)}
 
-# Streamlit UI
-st.title("Generate Caption from Image")
+
 
 uploaded_image = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])