Spaces:
Runtime error
Runtime error
Remove the ZeroGPU duration specification, to get past a related runtime error
Browse files
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(
|
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")
|