thamada commited on
Commit
bd8c9ee
·
verified ·
1 Parent(s): 8802e32

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -7,7 +7,7 @@ def runLLM ():
7
  tokenizer = AutoTokenizer.from_pretrained("cyberagent/open-calm-small")
8
 
9
  inputs = tokenizer("AIによって私達の暮らしは、", return_tensors="pt").to(model.device)
10
-
11
  with torch.no_grad():
12
  tokens = model.generate(
13
  **inputs,
@@ -20,6 +20,8 @@ def runLLM ():
20
  )
21
 
22
  output = tokenizer.decode(tokens[0], skip_special_tokens=True)
 
 
23
  return output
24
 
25
  def display_message():
 
7
  tokenizer = AutoTokenizer.from_pretrained("cyberagent/open-calm-small")
8
 
9
  inputs = tokenizer("AIによって私達の暮らしは、", return_tensors="pt").to(model.device)
10
+ """
11
  with torch.no_grad():
12
  tokens = model.generate(
13
  **inputs,
 
20
  )
21
 
22
  output = tokenizer.decode(tokens[0], skip_special_tokens=True)
23
+ """
24
+ output = "やあやあ"
25
  return output
26
 
27
  def display_message():