thamada commited on
Commit
90ab7f8
·
verified ·
1 Parent(s): bd8c9ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -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,
@@ -19,9 +19,12 @@ def runLLM ():
19
  pad_token_id=tokenizer.pad_token_id,
20
  )
21
 
 
22
  output = tokenizer.decode(tokens[0], skip_special_tokens=True)
23
  """
 
24
  output = "やあやあ"
 
25
  return output
26
 
27
  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,
 
19
  pad_token_id=tokenizer.pad_token_id,
20
  )
21
 
22
+ """
23
  output = tokenizer.decode(tokens[0], skip_special_tokens=True)
24
  """
25
+
26
  output = "やあやあ"
27
+
28
  return output
29
 
30
  def display_message():