Update app.py

app.py

@@ -1,17 +1,21 @@
 import gradio as gr
+import spaces
 import torch
 from PIL import Image
 from pathlib import Path
 from main import load_and_preprocess_image, genomic_plip_predictions, classify_tiles
 
+@spaces.GPU
 def run_load_and_preprocess_image(image_path, clip_processor_path):
  image_tensor = load_and_preprocess_image(image_path, clip_processor_path)
  return image_tensor
 
+@spaces.GPU
 def run_genomic_plip_predictions(image_tensor, model_path):
  pred_data = genomic_plip_predictions(image_tensor, model_path)
  return pred_data
 
+@spaces.GPU
 def run_classify_tiles(pred_data, model_path):
  output = classify_tiles(pred_data, model_path)
  return output
@@ -54,4 +58,4 @@ with gr.Blocks() as demo:
  gr.Markdown("## Example Images")
  gr.Examples(example_files, inputs=image_file)
 
-demo.launch()
+demo.launch()