Update app.py

model selection

app.py

@@ -45,6 +45,20 @@ elif page == "Invoices to Table":
         index=3,
         key="token_selector_sidebar"
     )
+        # Model selector
+    st.sidebar.markdown("### 🤖 Choose LLM Model")
+    selected_model = st.sidebar.radio(
+        "Which model to use?",
+        options=[
+            "meta-llama/llama-4-maverick-17b-128e-instruct",
+            "meta-llama/llama-4-scout-17b-16e-instruct",
+            "qwen/qwen3-32b",
+            "llama-3.3-70b-versatile",
+            "gemma2-9b-it"
+        ],
+        index=1,
+        key="model_selector_sidebar"
+    )
     
     uploaded_file = st.file_uploader("📤 Upload Invoice Image", type=["jpg", "jpeg", "png"])
     col1, col2 = st.columns([1.5, 1])  # col1 = processing, col2 = image
@@ -68,7 +82,7 @@ elif page == "Invoices to Table":
                 st.text_area("OCR Text", text, height=300)
 
             with st.spinner("📊 Extracting Table..."):
-                table_md = extract_table_from_text(text,max_tokens=selected_token_limit)
+                table_md = extract_table_from_text(text,max_tokens=selected_token_limit,model=selected_model)
 
             if st.button("🧠 Parse Table"):
                 try: