zeroMN commited on
Commit
63549c9
verified
1 Parent(s): feacfef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ def load_model():
12
  def infer(text):
13
  model, tokenizer = load_model()
14
  inputs = tokenizer(text, return_tensors="pt")
15
- outputs = model(**inputs)
16
  # 鏍规嵁妯″瀷杈撳嚭鐢熸垚鍝嶅簲
17
  return f"Predicted response for: {text}"
18
 
 
12
  def infer(text):
13
  model, tokenizer = load_model()
14
  inputs = tokenizer(text, return_tensors="pt")
15
+ outputs = model("zeroMN/zeroMN_SHMT")
16
  # 鏍规嵁妯″瀷杈撳嚭鐢熸垚鍝嶅簲
17
  return f"Predicted response for: {text}"
18