Florence-2-base-Castollux-v0.6

Running

xzuyn commited on Feb 6

Commit

93282d7

verified ·

1 Parent(s): 3e8d694

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,15 +12,15 @@ import torch
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 model = AutoModelForCausalLM.from_pretrained(
-    'PJMixers-Images/Florence-2-base-Castollux-v0.4',
     trust_remote_code=True
 ).eval()
 processor = AutoProcessor.from_pretrained(
-    'PJMixers-Images/Florence-2-base-Castollux-v0.4',
     trust_remote_code=True
 )
-TITLE = "# [PJMixers-Images/Florence-2-base-Castollux-v0.4](https://huggingface.co/PJMixers-Images/Florence-2-base-Castollux-v0.4)"
 @spaces.GPU

 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 model = AutoModelForCausalLM.from_pretrained(
+    'PJMixers-Images/Florence-2-base-Castollux-v0.5',
     trust_remote_code=True
 ).eval()
 processor = AutoProcessor.from_pretrained(
+    'PJMixers-Images/Florence-2-base-Castollux-v0.5',
     trust_remote_code=True
 )
+TITLE = "# [PJMixers-Images/Florence-2-base-Castollux-v0.5](https://huggingface.co/PJMixers-Images/Florence-2-base-Castollux-v0.5)"
 @spaces.GPU