Mohssinibra commited on
Commit
f1dc58b
·
verified ·
1 Parent(s): 93097a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ def forecast(csv_file):
25
 
26
  # Generate forecast
27
  with torch.no_grad():
28
- predictions = model.generate(**inputs, max_length=200, num_return_sequences=1)
29
 
30
  # Decode the generated forecast
31
  forecast_text = tokenizer.decode(predictions[0], skip_special_tokens=True)
 
25
 
26
  # Generate forecast
27
  with torch.no_grad():
28
+ predictions = model.generate(**inputs, max_length=2500, num_return_sequences=1)
29
 
30
  # Decode the generated forecast
31
  forecast_text = tokenizer.decode(predictions[0], skip_special_tokens=True)