lorenzoscottb commited on
Commit
ec8dc2b
β€’
1 Parent(s): 361a8a5

Update graph_utils.py

Browse files
Files changed (1) hide show
  1. graph_utils.py +1 -2
graph_utils.py CHANGED
@@ -25,9 +25,8 @@ def text_to_graph(text):
25
  pipe = pipeline(
26
  "text2text-generation",
27
  model=model_id,
28
- token=usr_tkn,
29
  max_length=300,
30
- min_length=10,
31
  )
32
 
33
  # generate text graph
 
25
  pipe = pipeline(
26
  "text2text-generation",
27
  model=model_id,
 
28
  max_length=300,
29
+ min_length=5,
30
  )
31
 
32
  # generate text graph