Update README.md
Browse files
README.md
CHANGED
|
@@ -82,8 +82,8 @@ Please do not use this for any commercial or factual purpose without re-training
|
|
| 82 |
from transformers import pipeline
|
| 83 |
|
| 84 |
qa = pipeline(
|
| 85 |
-
model="Avirallm/
|
| 86 |
-
tokenizer="Avirallm/
|
| 87 |
device="cuda" # or "cpu" if not using GPU
|
| 88 |
)
|
| 89 |
|
|
|
|
| 82 |
from transformers import pipeline
|
| 83 |
|
| 84 |
qa = pipeline(
|
| 85 |
+
model="Avirallm/Custom-Knowledge-LLM-Tony-Stark-Edition",
|
| 86 |
+
tokenizer="Avirallm/Custom-Knowledge-LLM-Tony-Stark-Edition",
|
| 87 |
device="cuda" # or "cpu" if not using GPU
|
| 88 |
)
|
| 89 |
|