ndebuhr commited on
Commit
1ac89b2
Β·
verified Β·
1 Parent(s): 3c30e6f

Remove the ZeroGPU duration specification, to get past a related runtime error

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -75,7 +75,7 @@ with beam.Pipeline(options=pipeline_options) as p:
75
 
76
 
77
  # Define a function to generate alert summaries using transformers and ZeroGPU
78
- @spaces.GPU(duration=300)
79
  def generate_summaries(alerts):
80
  huggingface_hub.login(token=os.environ["HUGGINGFACE_TOKEN"])
81
  device = torch.device("cuda")
 
75
 
76
 
77
  # Define a function to generate alert summaries using transformers and ZeroGPU
78
+ @spaces.GPU()
79
  def generate_summaries(alerts):
80
  huggingface_hub.login(token=os.environ["HUGGINGFACE_TOKEN"])
81
  device = torch.device("cuda")