donb-hf commited on
Commit
5a96b79
·
1 Parent(s): 6ce64a5

update timeout

Browse files
Files changed (1) hide show
  1. app.py +2 -32
app.py CHANGED
@@ -52,7 +52,7 @@ model.save_pretrained("merged_adapters")
52
 
53
  processor = PaliGemmaProcessor.from_pretrained(model_id)
54
 
55
- @spaces.GPU
56
  def infer(
57
  image: PIL.Image.Image,
58
  text: str,
@@ -313,10 +313,7 @@ def extract_objs(text, width, height, unique_labels=False):
313
  #########
314
 
315
  INTRO_TEXT="# 🔬🧠 CellVision AI -- Intelligent Cell Imaging Analysis 🤖🧫"
316
- IMAGE_PROMPT="""
317
- Describe the morphological characteristics and visible interactions between different cell types.
318
- Assess the biological context to identify signs of cancer and the presence of antigens.
319
- """
320
 
321
  with gr.Blocks(css="style.css") as demo:
322
  gr.Markdown(INTRO_TEXT)
@@ -331,41 +328,14 @@ with gr.Blocks(css="style.css") as demo:
331
 
332
  seg_btn = gr.Button("Submit")
333
  examples = [["./examples/cnmc1.bmp", "segment cancerous cells"],
334
- ["./examples/cnmc1.bmp", "segment healthy cells"],
335
- ["./examples/cnmc1.bmp", "detect cancerous cells"],
336
- ["./examples/cnmc1.bmp", "detect healthy cells"],
337
- ["./examples/cnmc2.bmp", "segment cancerous cells"],
338
- ["./examples/cnmc2.bmp", "segment healthy cells"],
339
  ["./examples/cnmc2.bmp", "detect cancerous cells"],
340
- ["./examples/cnmc2.bmp", "detect healthy cells"],
341
- ["./examples/cnmc3.bmp", "segment cancerous cells"],
342
  ["./examples/cnmc3.bmp", "segment healthy cells"],
343
- ["./examples/cnmc3.bmp", "detect cancerous cells"],
344
- ["./examples/cnmc3.bmp", "detect healthy cells"],
345
- ["./examples/cnmc4.bmp", "segment cancerous cells"],
346
- ["./examples/cnmc4.bmp", "segment healthy cells"],
347
- ["./examples/cnmc4.bmp", "detect cancerous cells"],
348
  ["./examples/cnmc4.bmp", "detect healthy cells"],
349
  ["./examples/cnmc5.bmp", "segment cancerous cells"],
350
- ["./examples/cnmc5.bmp", "segment healthy cells"],
351
- ["./examples/cnmc5.bmp", "detect cancerous cells"],
352
- ["./examples/cnmc5.bmp", "detect healthy cells"],
353
- ["./examples/cnmc6.bmp", "segment cancerous cells"],
354
- ["./examples/cnmc6.bmp", "segment healthy cells"],
355
  ["./examples/cnmc6.bmp", "detect cancerous cells"],
356
- ["./examples/cnmc6.bmp", "detect healthy cells"],
357
- ["./examples/cnmc7.bmp", "segment cancerous cells"],
358
  ["./examples/cnmc7.bmp", "segment healthy cells"],
359
- ["./examples/cnmc7.bmp", "detect cancerous cells"],
360
- ["./examples/cnmc7.bmp", "detect healthy cells"],
361
- ["./examples/cnmc8.bmp", "segment cancerous cells"],
362
- ["./examples/cnmc8.bmp", "segment healthy cells"],
363
- ["./examples/cnmc8.bmp", "detect cancerous cells"],
364
  ["./examples/cnmc8.bmp", "detect healthy cells"],
365
  ["./examples/cnmc9.bmp", "segment cancerous cells"],
366
- ["./examples/cnmc9.bmp", "segment healthy cells"],
367
- ["./examples/cnmc9.bmp", "detect cancerous cells"],
368
- ["./examples/cnmc9.bmp", "detect healthy cells"],
369
  ["./examples/cart1.jpg", "segment cells"],
370
  ["./examples/cart1.jpg", "detect cells"],
371
  ["./examples/cart2.jpg", "segment cells"],
 
52
 
53
  processor = PaliGemmaProcessor.from_pretrained(model_id)
54
 
55
+ @spaces.GPU(duration=120)
56
  def infer(
57
  image: PIL.Image.Image,
58
  text: str,
 
313
  #########
314
 
315
  INTRO_TEXT="# 🔬🧠 CellVision AI -- Intelligent Cell Imaging Analysis 🤖🧫"
316
+ IMAGE_PROMPT="Assess the biological context to identify signs of cancer and the presence of antigens."
 
 
 
317
 
318
  with gr.Blocks(css="style.css") as demo:
319
  gr.Markdown(INTRO_TEXT)
 
328
 
329
  seg_btn = gr.Button("Submit")
330
  examples = [["./examples/cnmc1.bmp", "segment cancerous cells"],
 
 
 
 
 
331
  ["./examples/cnmc2.bmp", "detect cancerous cells"],
 
 
332
  ["./examples/cnmc3.bmp", "segment healthy cells"],
 
 
 
 
 
333
  ["./examples/cnmc4.bmp", "detect healthy cells"],
334
  ["./examples/cnmc5.bmp", "segment cancerous cells"],
 
 
 
 
 
335
  ["./examples/cnmc6.bmp", "detect cancerous cells"],
 
 
336
  ["./examples/cnmc7.bmp", "segment healthy cells"],
 
 
 
 
 
337
  ["./examples/cnmc8.bmp", "detect healthy cells"],
338
  ["./examples/cnmc9.bmp", "segment cancerous cells"],
 
 
 
339
  ["./examples/cart1.jpg", "segment cells"],
340
  ["./examples/cart1.jpg", "detect cells"],
341
  ["./examples/cart2.jpg", "segment cells"],