J-LAB commited on
Commit
3b56f9e
·
verified ·
1 Parent(s): ca24932

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ model_id = 'J-LAB/Florence_2_B_FluxiAI_Product_Caption'
11
  model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True).to("cuda").eval()
12
  processor = AutoProcessor.from_pretrained(model_id, trust_remote_code=True)
13
 
14
- DESCRIPTION = "# Product Describe by Fluxi IA\n### Base Model [Florence-2] (https://huggingface.co/microsoft/Florence-2-large) ]"
15
 
16
  @spaces.GPU
17
  def run_example(task_prompt, image):
 
11
  model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True).to("cuda").eval()
12
  processor = AutoProcessor.from_pretrained(model_id, trust_remote_code=True)
13
 
14
+ DESCRIPTION = "# Product Describe by Fluxi IA\n### Base Model [Florence-2] (https://huggingface.co/microsoft/Florence-2-large)"
15
 
16
  @spaces.GPU
17
  def run_example(task_prompt, image):