Spaces:

Jalalkhan912
/

DialogSum

Sleeping

Jalalkhan912 commited on Oct 25

Commit

441ee20

•

1 Parent(s): c3b17cb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -35,7 +35,7 @@ What was going on?
     output = tokenizer.decode(
         model.generate(
             inputs["input_ids"],
-            max_new_tokens=80,
         )[0],
         skip_special_tokens=True
     )
@@ -77,7 +77,7 @@ def one_shot(example_indices_full,my_example):
   output = tokenizer.decode(
       model.generate(
           inputs["input_ids"],
-          max_new_tokens=80,
       )[0],
       skip_special_tokens=True
   )
@@ -89,7 +89,7 @@ def few_shot(example_indices_full_few_shot,my_example):
   output = tokenizer.decode(
       model.generate(
           inputs["input_ids"],
-          max_new_tokens=80,
       )[0],
       skip_special_tokens=True
   )

     output = tokenizer.decode(
         model.generate(
             inputs["input_ids"],
+            max_new_tokens=50
         )[0],
         skip_special_tokens=True
     )
   output = tokenizer.decode(
       model.generate(
           inputs["input_ids"],
+          max_new_tokens=50
       )[0],
       skip_special_tokens=True
   )
   output = tokenizer.decode(
       model.generate(
           inputs["input_ids"],
+          max_new_tokens=50
       )[0],
       skip_special_tokens=True
   )