sanot commited on
Commit
a276e06
1 Parent(s): d34083a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -6,6 +6,8 @@ from huggingface_hub import login
6
 
7
  from transformers import pipeline
8
  login(token=os.getenv("access_key"))
 
 
9
  messages1 = [
10
  {"role": "user", "content": "Who are you?"},
11
  ]
@@ -38,7 +40,6 @@ def respond(
38
 
39
  response = ""
40
 
41
- client = pipeline("text-generation", model="google/recurrentgemma-2b-it")
42
 
43
  token = client(
44
  messages, max_new_tokens= max_tokens
 
6
 
7
  from transformers import pipeline
8
  login(token=os.getenv("access_key"))
9
+ client = pipeline("text-generation", model="google/recurrentgemma-2b-it")
10
+
11
  messages1 = [
12
  {"role": "user", "content": "Who are you?"},
13
  ]
 
40
 
41
  response = ""
42
 
 
43
 
44
  token = client(
45
  messages, max_new_tokens= max_tokens