Update app.py

app.py

@@ -6,6 +6,12 @@ from fpdf import FPDF
 import tempfile
 from io import BytesIO
 from transformers import AutoTokenizer, AutoModelForCausalLM
+import torch
+
+# ---------------------------------------------------------------------
+# MUST BE FIRST STREAMLIT CALL
+# ---------------------------------------------------------------------
+st.set_page_config(page_title="Smart Log Copilot", layout="wide")
 
 # ------------------ MODEL LOADING ------------------
 MODEL_NAME = "Qwen/Qwen2.5-1.5B-Instruct"
@@ -15,7 +21,7 @@ def load_llm():
     model = AutoModelForCausalLM.from_pretrained(
         MODEL_NAME,
         device_map="auto",
-        torch_dtype="auto"
+        torch_dtype=torch.float16 if torch.cuda.is_available() else "auto"
     )
     return tokenizer, model
 
@@ -139,7 +145,6 @@ def build_pdf(risk_icon, risk_label, summary, anomalies):
 
 
 # ------------------ STREAMLIT UI ------------------
-st.set_page_config(page_title="Smart Log Copilot", layout="wide")
 st.title("🔍 Smart Log Copilot (CSV-powered LLM Demo)")
 
 uploaded = st.file_uploader("Upload CSV log file", type=["csv"])