habdine commited on
Commit
9cbac99
1 Parent(s): 9a9a02f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ def generate(
51
  num_beams=1,
52
  repetition_penalty=repetition_penalty,
53
  )
54
- t = Thread(target=model.generate, kwargs=generate_kwargs)
55
  t.start()
56
 
57
  outputs = []
 
51
  num_beams=1,
52
  repetition_penalty=repetition_penalty,
53
  )
54
+ t = Thread(target=model.generate_protein_description, kwargs=generate_kwargs)
55
  t.start()
56
 
57
  outputs = []