trust remote code

README.md

@@ -12,3 +12,9 @@ short_description: Hold me closer, tiny coder!
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+Test query:
+
+
+```
+write a simple Fizz Buzz function in Python.

app.py

@@ -1,5 +1,3 @@
-# app.py
-
 import os
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForCausalLM
@@ -20,11 +18,12 @@ def load_model(model_size: str = "32B"):
     
     model_id = model_map.get(model_size, "Qwen/Qwen-7B")  # default to 7B if size not found
     
-    tokenizer = AutoTokenizer.from_pretrained(model_id)
+    tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
     model = AutoModelForCausalLM.from_pretrained(
         model_id,
         torch_dtype=torch.float16,
-        device_map="auto"
+        device_map="auto",
+        trust_remote_code=True
     )
     return model, tokenizer