Nerva1228 commited on
Commit
eb96fb2
·
verified ·
1 Parent(s): 289ea90

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
10
  model = AutoModelForCausalLM.from_pretrained("MiaoshouAI/Florence-2-base-PromptGen-v1.5", trust_remote_code=True).to(device)
11
  processor = AutoProcessor.from_pretrained("MiaoshouAI/Florence-2-base-PromptGen-v1.5", trust_remote_code=True)
12
 
13
- @spaces.GPU(duration=180)
14
  def infer(prompt, image, request: gr.Request):
15
  if request:
16
  print("请求头字典:", request.headers)
 
10
  model = AutoModelForCausalLM.from_pretrained("MiaoshouAI/Florence-2-base-PromptGen-v1.5", trust_remote_code=True).to(device)
11
  processor = AutoProcessor.from_pretrained("MiaoshouAI/Florence-2-base-PromptGen-v1.5", trust_remote_code=True)
12
 
13
+ @spaces.GPU(duration=200)
14
  def infer(prompt, image, request: gr.Request):
15
  if request:
16
  print("请求头字典:", request.headers)