Update app.py

app.py

@@ -45,7 +45,7 @@ def split_text(text, chunk_size=1024):
         yield ' '.join(words[i:i + chunk_size])
 
 # Function to classify text using LED model
-@spaces.GPU
+@spaces.GPU(duration=120)
 def classify_text(text):
     try:
         return classifier(text)[0]['label']
@@ -53,7 +53,7 @@ def classify_text(text):
         return "Unable to classify"
 
 # Function to summarize text using the summarizer model
-@spaces.GPU
+@spaces.GPU(duration=120)
 def summarize_text(text, max_length=100, min_length=30):
     try:
         return summarizer(text, max_length=max_length, min_length=min_length, do_sample=False)[0]['summary_text']
@@ -61,7 +61,7 @@ def summarize_text(text, max_length=100, min_length=30):
         return "Unable to summarize"
 
 # Function to extract a title-like summary from the beginning of the text
-@spaces.GPU
+@spaces.GPU(duration=120)
 def extract_title(text, max_length=20):
     try:
         return summarizer(text, max_length=max_length, min_length=5, do_sample=False)[0]['summary_text']
@@ -72,7 +72,7 @@ def extract_title(text, max_length=20):
 # output_folder_path = '/content/drive/My Drive/path_to_output'  # Adjust this to your actual path
 
 # Define the Gradio interface for file upload and download
-@spaces.GPU
+@spaces.GPU(duration=120)
 def process_files(pdf_files):
     data = []
     for pdf_file in pdf_files: