Update app.py
Browse files
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 |
-
|
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,
|