.

app.py

@@ -125,7 +125,6 @@ TECHNIQUE_LIST = [
 ]
 
 
-@spaces.GPU(duration=120)
 @torch.no_grad()
 def phrase_log_prob(prompt, phrase):
     """Calculate log probability of a phrase given a prompt using the language model."""
@@ -250,6 +249,7 @@ def campaign_actor_probs(campaigns, actors, prompt_template="is conducted by"):
     return pd.DataFrame.from_dict(rows, orient="index")
 
 
+@spaces.GPU(duration=120)
 def generate_actor_heatmap(c_list, a_list, actor_prompt_template):
     """Generate Campaign-Actor association heatmap with probability visualization."""
     try:
@@ -314,6 +314,7 @@ def generate_actor_heatmap(c_list, a_list, actor_prompt_template):
         return fig
 
 
+@spaces.GPU(duration=120)
 def generate_technique_heatmap(c_list, t_list, technique_prompt_template, technique_neg_template):
     """Generate Campaign-Technique association heatmap with binary scoring visualization."""
     try: