Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -33,7 +33,7 @@ def create_response_fine_tuned(input_str):
|
|
33 |
#output_str = tokenizer.decode(model.generate(**tokenizer("What are John West's hobbies?"+tokenizer.eos_token,return_tensors="pt",max_length=200))[0])
|
34 |
output_str = tokenizer.decode(fine_tuned_model.generate(**tokenizer(input_str+tokenizer.eos_token,return_tensors="pt",max_length=200))[0])
|
35 |
return (output_str)
|
36 |
-
interface1 = gr.Interface(fn=create_response_original, inputs=["text",gr.inputs.Number(label="Enter a whole number")], outputs="text", title="Original")
|
37 |
interface2 = gr.Interface(fn=create_response_fine_tuned, inputs="text", outputs="text", title="Fine Tuned")
|
38 |
demo = gr.TabbedInterface([interface1, interface2], ["Original", "Fine Tuned"])
|
39 |
# with gr.Blocks() as demo:
|
|
|
33 |
#output_str = tokenizer.decode(model.generate(**tokenizer("What are John West's hobbies?"+tokenizer.eos_token,return_tensors="pt",max_length=200))[0])
|
34 |
output_str = tokenizer.decode(fine_tuned_model.generate(**tokenizer(input_str+tokenizer.eos_token,return_tensors="pt",max_length=200))[0])
|
35 |
return (output_str)
|
36 |
+
interface1 = gr.Interface(fn=create_response_original, inputs=[gr.Textbox(label="input text here", lines=3),gr.inputs.Number(label="Enter a whole number")], outputs="text", title="Original")
|
37 |
interface2 = gr.Interface(fn=create_response_fine_tuned, inputs="text", outputs="text", title="Fine Tuned")
|
38 |
demo = gr.TabbedInterface([interface1, interface2], ["Original", "Fine Tuned"])
|
39 |
# with gr.Blocks() as demo:
|