Update app.py

app.py

@@ -13,8 +13,7 @@ subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENT
 
 model = AutoModelForCausalLM.from_pretrained(
     'PJMixers-Dev/Florence-2-base-danbooru2022-316k',
-    trust_remote_code=True,
-    torch_dtype=torch.float32
+    trust_remote_code=True
 ).eval()
 processor = AutoProcessor.from_pretrained(
     'PJMixers-Dev/Florence-2-base-danbooru2022-316k',
@@ -34,7 +33,7 @@ def process_image(image):
     if image.mode != "RGB":
         image = image.convert("RGB")
     
-    inputs = processor(text="<CAPTION>", images=image, return_tensors="pt").to(torch.float32)
+    inputs = processor(text="<CAPTION>", images=image, return_tensors="pt")
     generated_ids = model.generate(
         input_ids=inputs["input_ids"],
         pixel_values=inputs["pixel_values"],