Update app.py
Browse files
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
|
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")
|