wasmdashai commited on
Commit
b2e1dba
·
verified ·
1 Parent(s): e60eceb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
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= {}