Update app.py

app.py

@@ -20,7 +20,7 @@ model = YOLOv10.from_pretrained('jameslahm/yolov10x').to(device)
 
 model_id = "meta-llama/Llama-3.2-11B-Vision-Instruct"
 
-model = MllamaForConditionalGeneration.from_pretrained(
+model_code = MllamaForConditionalGeneration.from_pretrained(
     model_id,
     torch_dtype=torch.bfloat16,
     device_map="auto",
@@ -104,7 +104,7 @@ def generate_journal_infographics(journal):
     model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
     
     # Generate the documentation
-    generated_ids = model.generate(**model_inputs, max_new_tokens=4000)
+    generated_ids = model_code.generate(**model_inputs, max_new_tokens=4000)
     generated_ids = [output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)]
     documentation = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
     print(documentation)