Update app.py

app.py

@@ -14,10 +14,12 @@ model_name = 'scb10x/llama-3-typhoon-v1.5-8b-instruct-vision-preview'
 def load_model():
     model = AutoModelForCausalLM.from_pretrained(
         model_name,
+        revision='main',  # Or a specific commit hash
         torch_dtype=torch.float16 if device == 'cuda' else torch.float32,
         device_map='auto',
         trust_remote_code=True
     )
+
     return model
 
 model = load_model()