Update finetune.py

finetune.py

@@ -180,21 +180,19 @@ def train(input_text):
     repo.push_to_hub(OUTPUT_DIR, commit_message="Ft model")
 
 
-# iface = gr.Interface(
-#     fn=train,
-#     inputs=gr.inputs.Textbox(label="Input text"),
-#     outputs=gr.outputs.Textbox(label="Output length"),
-#     title="Training Interface",
-#     description="Enter some text and click the button to start training.",
-#     theme="default",
-#     layout="vertical",
-#     allow_flagging=False,
-#     allow_screenshot=False
-# )
-
-if __name__ == '__main__':
-    # iface.launch(share=True)
-    train("Oi")
+iface = gr.Interface(
+    fn=train,
+    inputs=gr.inputs.Textbox(label="Input text"),
+    outputs=gr.outputs.Textbox(label="Output length"),
+    title="Training Interface",
+    description="Enter some text and click the button to start training.",
+    theme="default",
+    layout="vertical",
+    allow_flagging=False,
+    allow_screenshot=False
+)
+iface.launch(share=True)
+