Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 os.environ["GRADIO_TEMP_DIR"] = "./tmp"
 
 import sys
+import space
 import torch
 import torchvision
 import gradio as gr
@@ -28,6 +29,7 @@ id_to_names = {
     9: 'formula_caption'
 }
 
+@spaces.GPU
 def recognize_image(input_img, conf_threshold, iou_threshold):
     det_res = model.predict(
         input_img,