habdine commited on
Commit
7042d65
1 Parent(s): 4a0a200

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ def generate(
42
 
43
  streamer = TextIteratorStreamer(tokenizer, timeout=20.0, skip_prompt=True, skip_special_tokens=True)
44
  generate_kwargs = dict(
45
- protein_sequence=message,
46
  tokenizer=tokenizer,
47
  device=device,
48
  streamer=streamer,
 
42
 
43
  streamer = TextIteratorStreamer(tokenizer, timeout=20.0, skip_prompt=True, skip_special_tokens=True)
44
  generate_kwargs = dict(
45
+ protein_pdbID=message,
46
  tokenizer=tokenizer,
47
  device=device,
48
  streamer=streamer,