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,
|