Update app.py

app.py

@@ -8,12 +8,17 @@ model_path = "sbintuitions/sarashina2-vision-8b"
 
 print("モデルを読み込んでいます...")
 processor = AutoProcessor.from_pretrained(model_path, trust_remote_code=True)
+
+# デバイスの設定
+device = "cuda" if torch.cuda.is_available() else "cpu"
+print(f"使用デバイス: {device}")
+
 model = AutoModelForCausalLM.from_pretrained(
     model_path,
-    device_map="auto",
-    torch_dtype=torch.float16,
+    torch_dtype=torch.float16 if device == "cuda" else torch.float32,
     trust_remote_code=True,
 )
+model = model.to(device)
 print("モデルの読み込みが完了しました！")
 
 def describe_image(image):