Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,7 +4,14 @@ import torch
|
|
| 4 |
from transformers import AutoTokenizer,VitsModel
|
| 5 |
import os
|
| 6 |
import numpy as np
|
| 7 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
token=os.environ.get("key_")
|
| 9 |
tokenizer = AutoTokenizer.from_pretrained("wasmdashai/vtk",token=token)
|
| 10 |
models= {}
|
|
|
|
| 4 |
from transformers import AutoTokenizer,VitsModel
|
| 5 |
import os
|
| 6 |
import numpy as np
|
| 7 |
+
# Use a pipeline as a high-level helper
|
| 8 |
+
from transformers import pipeline
|
| 9 |
+
|
| 10 |
+
pipe = pipeline("text-generation", model="wasmdashai/Seed-Coder-8B-Instruct-V1")
|
| 11 |
+
messages = [
|
| 12 |
+
{"role": "user", "content": "Who are you?"},
|
| 13 |
+
]
|
| 14 |
+
pipe(messages)
|
| 15 |
token=os.environ.get("key_")
|
| 16 |
tokenizer = AutoTokenizer.from_pretrained("wasmdashai/vtk",token=token)
|
| 17 |
models= {}
|