dsfdfsghgf commited on
Commit
ce5f258
·
verified ·
1 Parent(s): 12d6b61

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -40,11 +40,11 @@ except Exception as e:
40
  # Inferenz durchführen
41
  try:
42
  with torch.no_grad():
43
- outputs = model.generate(input_ids=inputs["input_ids"].to(device),
44
- attention_mask=inputs["attention_mask"].to(device),
45
- max_length=256,
46
- num_return_sequences=1)
47
- print("Inferenz abgeschlossen.")
48
  except Exception as e:
49
  print(f"Fehler bei der Inferenz: {e}")
50
 
 
40
  # Inferenz durchführen
41
  try:
42
  with torch.no_grad():
43
+ outputs = model.generate(
44
+ input_ids=inputs["input_ids"].to(device),
45
+ max_length=50 # Versuche mit einer kleineren Länge
46
+ )
47
+ print("Inferenz abgeschlossen.")
48
  except Exception as e:
49
  print(f"Fehler bei der Inferenz: {e}")
50