Update app.py

app.py

@@ -30,7 +30,7 @@ def summarize_pdf(pdf_file, model):
             text += page.get_text()
     return summarize_text(text, model)
 
-@spaces.GPU(duration=60)
+@spaces.GPU(duration=30)
 def summarize(input_type, input_text, uploaded_file):
     try:
         if input_type == "Text":
@@ -38,9 +38,15 @@ def summarize(input_type, input_text, uploaded_file):
         else:
             summary = summarize_pdf(uploaded_file, summarizer)
         return summary
+    except RuntimeError as e:
+        if "CUDA out of memory" in str(e):
+            return "There was a problem summarizing the text due to insufficient GPU memory. Please try again with a smaller input."
+        else:
+            return "There was a problem summarizing the text. Please try again later."
     except Exception as e:
         return "There was a problem summarizing the text. Please try again later."
 
+
 # Define the footer
 footer = """
 <div style="text-align: center; margin-top: 20px;">