Update app.py

app.py

@@ -4,12 +4,6 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 from typing import Optional
 
-# 🧩 LangSmith Integration (Optional)
-os.environ["LANGCHAIN_TRACING_V2"] = "true"
-os.environ["LANGCHAIN_PROJECT"] = "Qwen_4B_Chatbot"
-os.environ["LANGCHAIN_API_KEY"] = "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"
-
-
 app = FastAPI(title="🤖 Qwen 4B AI Chatbot")
 
 # ✅ Set writable cache directory
@@ -26,7 +20,7 @@ tokenizer = AutoTokenizer.from_pretrained(
 
 model = AutoModelForCausalLM.from_pretrained(
     model_name,
-    torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
+    dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
     device_map="auto",
     trust_remote_code=True
 )
@@ -109,7 +103,6 @@ def home():
     }
 
 @app.post("/chat")
-@traceable
 async def chat(
     query: str = Form(...),
     max_tokens: int = Form(512),
@@ -128,6 +121,7 @@ async def chat(
     - "Calculate 15% of 2500 and explain compound interest"
     - "Search for the latest AI news and summarize the top 3 trends"
     - "Write Python code to sort a list and explain how it works"
+    - "What's the population of Belarus and its GDP?"
     """
     try:
         if not query or len(query.strip()) == 0: