Update app.py

app.py

@@ -9,13 +9,13 @@ import json
 # Login using secret (secure, no hardcode)
 login(os.environ["HF_TOKEN"])
 
-# Model setup (loads once on Space startup)
-model_id = "agarkovv/CryptoTrader-LM"
-base_model_id = "mistralai/Ministral-8B-Instruct-2410"
+# Model setup (loads once on Space startup; switched to Llama 3 base)
+model_id = "agarkovv/CryptoTrader-LM"  # Keep PEFT adapter if compatible; otherwise fine-tune on Llama
+base_model_id = "meta-llama/Meta-Llama-3-8B-Instruct"
 MAX_LENGTH = 32768
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"  # Use GPU if available (ZeroGPU on HF)
 
-# Workaround: Download model files, edit adapter_config.json to remove 'model_type'
+# Workaround: Download model files, edit adapter_config.json to remove 'model_type' if present
 local_dir = snapshot_download(repo_id=model_id)
 config_path = os.path.join(local_dir, "adapter_config.json")
 with open(config_path, 'r') as f:
@@ -62,8 +62,8 @@ demo = gr.Interface(
     fn=predict_trading_decision,
     inputs=gr.Textbox(label="Input Prompt (News + Prices)"),
     outputs=gr.Textbox(label="Trading Decision"),
-    title="CryptoTrader-LM MCP Tool",
-    description="Predict buy/sell/hold for BTC/ETH."
+    title="CryptoTrader-LM with Llama MCP Tool",
+    description="Predict buy/sell/hold for BTC/ETH using Llama 3 base."
 )
 
 # Launch with MCP support