Spaces:
Sleeping
Sleeping
two interfaces using tab
Browse files
app.py
CHANGED
@@ -32,10 +32,10 @@ def create_response_fine_tuned(input_str):
|
|
32 |
#output_str = tokenizer.decode(model.generate(**tokenizer("What are John West's hobbies?"+tokenizer.eos_token,return_tensors="pt",max_length=200))[0])
|
33 |
output_str = tokenizer.decode(fine_tuned_model.generate(**tokenizer(input_str+tokenizer.eos_token,return_tensors="pt",max_length=200))[0])
|
34 |
return (output_str)
|
35 |
-
interface1 = gr.Interface(fn=create_response_original, inputs="text", outputs="text")
|
36 |
-
interface2 = gr.Interface(fn=create_response_fine_tuned, inputs="text", outputs="text")
|
37 |
-
|
38 |
-
|
39 |
-
|
40 |
-
|
41 |
demo.launch()
|
|
|
32 |
#output_str = tokenizer.decode(model.generate(**tokenizer("What are John West's hobbies?"+tokenizer.eos_token,return_tensors="pt",max_length=200))[0])
|
33 |
output_str = tokenizer.decode(fine_tuned_model.generate(**tokenizer(input_str+tokenizer.eos_token,return_tensors="pt",max_length=200))[0])
|
34 |
return (output_str)
|
35 |
+
interface1 = gr.Interface(fn=create_response_original, inputs="text", outputs="text", title="Original")
|
36 |
+
interface2 = gr.Interface(fn=create_response_fine_tuned, inputs="text", outputs="text", title="Fine Tuned")
|
37 |
+
demo = gr.TabbedInterface([interface1, interface2], ["Original", "Fine Tuned"])
|
38 |
+
# with gr.Blocks() as demo:
|
39 |
+
# with gr.Row():
|
40 |
+
#
|
41 |
demo.launch()
|