crystal99 commited on
Commit
fc3f78f
·
verified ·
1 Parent(s): f0ffc11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -42,10 +42,10 @@ def generate_text(prompt):
42
  inputs = tokenizer(f"{prompt}", return_tensors="pt").to(device)
43
  outputs = model.generate(inputs['input_ids'], max_length=100, num_return_sequences=1, do_sample=False)
44
  generated_text = tokenizer.decode(outputs[0], skip_special_tokens=False)
45
- result2 = generated_text.split("<|ENDOFTEXT|>")
46
- finalRes = result2[0].split("<|BOT|>")
47
  print(generated_text)
48
- return finalRes[-1]
49
 
50
  # Set up the Gradio interface
51
  iface = gr.Interface(fn=generate_text, inputs="text", outputs="text", title="Text Generator using Fine-Tuned Model")
 
42
  inputs = tokenizer(f"{prompt}", return_tensors="pt").to(device)
43
  outputs = model.generate(inputs['input_ids'], max_length=100, num_return_sequences=1, do_sample=False)
44
  generated_text = tokenizer.decode(outputs[0], skip_special_tokens=False)
45
+ result2 = generated_text #.split("<|ENDOFTEXT|>")
46
+ finalRes = result2 #[0].split("<|BOT|>")
47
  print(generated_text)
48
+ return finalRes #[-1]
49
 
50
  # Set up the Gradio interface
51
  iface = gr.Interface(fn=generate_text, inputs="text", outputs="text", title="Text Generator using Fine-Tuned Model")