injilashah commited on
Commit
2e9098c
·
verified ·
1 Parent(s): 3f51521

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -21,8 +21,8 @@ def Sentence_Commpletion(model_name, input):
21
  outputs = model.generate(input.inputid, max_length=30, num_return_sequences=1)
22
  elif model_name == "Gemma":
23
  tokenizer, model = g_tokenizer, g_model
24
- inputid = Tokenizer(input, return_tensors="pt")
25
- outputs = Model.generate(**inputid, max_new_tokens=20)
26
  return tokenizer.decode(outputs[0])
27
 
28
 
 
21
  outputs = model.generate(input.inputid, max_length=30, num_return_sequences=1)
22
  elif model_name == "Gemma":
23
  tokenizer, model = g_tokenizer, g_model
24
+ inputid = tokenizer(input, return_tensors="pt")
25
+ outputs = model.generate(**inputid, max_new_tokens=20)
26
  return tokenizer.decode(outputs[0])
27
 
28