Vasudevakrishna commited on
Commit
f669094
·
verified ·
1 Parent(s): 6fca767

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ model.load_adapter(peft_model_folder)
18
  def generate_text(input_text, max_length):
19
  pipe = pipeline(task="text-generation",model=model,tokenizer=tokenizer, max_length=max_length)
20
  result = pipe(f"<s>[INST] {input_text} [/INST]")
21
- return_answer = (result[0]['generated_text'])
22
  return return_answer
23
 
24
  # Create a Gradio interface
 
18
  def generate_text(input_text, max_length):
19
  pipe = pipeline(task="text-generation",model=model,tokenizer=tokenizer, max_length=max_length)
20
  result = pipe(f"<s>[INST] {input_text} [/INST]")
21
+ return_answer = result[0]['generated_text']
22
  return return_answer
23
 
24
  # Create a Gradio interface