rubendsm commited on
Commit
44ca29e
verified
1 Parent(s): 3f67596

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -27,8 +27,10 @@ def summarize_long_text(input_text, temperature=0.40, top_k=55, top_p=0.75, num_
27
  part_summaries = []
28
  for part in parts:
29
  input_ids = tokenizer.encode(part, return_tensors="pt", max_length=max_input_length, truncation=True)
 
 
30
  output = model.generate(
31
- input_ids.to(device), # IDs de entrada codificados
32
  max_length=max_output_length, # Comprimento m谩ximo do texto gerado
33
  num_return_sequences=1, # N煤mero de sequ锚ncias de retorno
34
  do_sample=True, # Habilita a amostragem aleat贸ria
 
27
  part_summaries = []
28
  for part in parts:
29
  input_ids = tokenizer.encode(part, return_tensors="pt", max_length=max_input_length, truncation=True)
30
+ # Move os dados de entrada para o dispositivo adequado
31
+ input_ids = input_ids.to(device)
32
  output = model.generate(
33
+ input_ids, # IDs de entrada codificados
34
  max_length=max_output_length, # Comprimento m谩ximo do texto gerado
35
  num_return_sequences=1, # N煤mero de sequ锚ncias de retorno
36
  do_sample=True, # Habilita a amostragem aleat贸ria