C2MV commited on
Commit
e80a6ea
1 Parent(s): f5a37da

Update interface.py

Browse files
Files changed (1) hide show
  1. interface.py +2 -1
interface.py CHANGED
@@ -22,7 +22,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_path)
22
  model = AutoModelForCausalLM.from_pretrained(model_path)
23
  # No movemos el modelo al dispositivo aquí
24
 
25
- @spaces.GPU(duration=300)
26
  def generate_analysis(prompt, max_length=MAX_LENGTH, device=None):
27
  try:
28
  if device is None:
@@ -60,6 +60,7 @@ def parse_bounds(bounds_str, num_params):
60
  upper_bounds = [np.inf] * num_params
61
  return lower_bounds, upper_bounds
62
 
 
63
  def process_and_plot(
64
  file,
65
  biomass_eq1, biomass_eq2, biomass_eq3,
 
22
  model = AutoModelForCausalLM.from_pretrained(model_path)
23
  # No movemos el modelo al dispositivo aquí
24
 
25
+ @spaces.GPU(duration=100)
26
  def generate_analysis(prompt, max_length=MAX_LENGTH, device=None):
27
  try:
28
  if device is None:
 
60
  upper_bounds = [np.inf] * num_params
61
  return lower_bounds, upper_bounds
62
 
63
+ @spaces.GPU(duration=100)
64
  def process_and_plot(
65
  file,
66
  biomass_eq1, biomass_eq2, biomass_eq3,