Update app.py

app.py

@@ -6,8 +6,9 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 
 # Modell und Tokenizer laden
 model = AutoModel.from_pretrained(
-    model_name, 
-    device_map="auto", 
+    model_name,
+    revision="main",  # Hier den Namen des Branches oder Commit-Hash einfügen
+    device_map=device,
     torch_dtype=torch.bfloat16,
     trust_remote_code=True,
 ).eval()