a

app.py

@@ -145,7 +145,7 @@ def create_messages(review_fields, paper_text):
     ]
     return messages
 
-@spaces.GPU()
+@spaces.GPU(60)
 def convert_file(filepath):
     full_text, images, out_metadata = convert_single_pdf(
             filepath, model_refs, metadata=metadata, max_pages=MAX_PAGES
@@ -170,6 +170,10 @@ def process_file(file):
     except spaces.zero.gradio.HTMLError as e:
         print(e)
         return "Error. GPU quota exceeded. Please return later."
+    except gradio.exceptions.Error as e:
+        if 'GPU task aborted' in str(e):
+            print(e)
+            return 'GPU task aborted'
     except Exception as e:
         print(traceback.format_exc())
         print(f"Error converting {filepath}: {e}")