Update app.py

app.py

@@ -30,7 +30,9 @@ REVISION = "ceaf371f01ef66192264811b390bccad475a4f02"
 
 # Florence-2 로드
 device = "cuda" if torch.cuda.is_available() else "cpu"
-florence_model = AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-base', revision = REVISION, trust_remote_code=True).to(device).eval()
+florence_model = AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-base', revision = REVISION, trust_remote_code=True, torch_dtype=torch.float16)
+florence_model.to("cpu")
+florence_model.eval()
 florence_processor = AutoProcessor.from_pretrained('microsoft/Florence-2-base', revision = REVISION, trust_remote_code=True)
 
 # Stable Diffusion TurboX 로드