Update app.py

app.py

@@ -57,12 +57,13 @@ class SmolLM2ForCausalLM(PreTrainedModel):
         super().__init__(config)
         self.config = config
         
-        # Load the model weights directly from the checkpoint
+        # Initialize model weights from your checkpoint
         self.model = AutoModelForCausalLM.from_pretrained(
-            "meta-llama/Llama-2-7b-hf",
+            "jatingocodeo/SmolLM2",
             config=config,
             torch_dtype=torch.float16,
-            low_cpu_mem_usage=True
+            low_cpu_mem_usage=True,
+            trust_remote_code=True
         )
         
     def forward(self, input_ids=None, attention_mask=None, labels=None, **kwargs):