gustavoaq commited on
Commit
405873b
1 Parent(s): adfcdb4

Update finetune.py

Browse files
Files changed (1) hide show
  1. finetune.py +13 -12
finetune.py CHANGED
@@ -180,20 +180,21 @@ def train(input_text):
180
  repo.push_to_hub(OUTPUT_DIR, commit_message="Ft model")
181
 
182
 
183
- iface = gr.Interface(
184
- fn=train,
185
- inputs=gr.inputs.Textbox(label="Input text"),
186
- outputs=gr.outputs.Textbox(label="Output length"),
187
- title="Training Interface",
188
- description="Enter some text and click the button to start training.",
189
- theme="default",
190
- layout="vertical",
191
- allow_flagging=False,
192
- allow_screenshot=False
193
- )
194
 
195
  if __name__ == '__main__':
196
- iface.launch(share=True)
 
197
 
198
 
199
 
 
180
  repo.push_to_hub(OUTPUT_DIR, commit_message="Ft model")
181
 
182
 
183
+ # iface = gr.Interface(
184
+ # fn=train,
185
+ # inputs=gr.inputs.Textbox(label="Input text"),
186
+ # outputs=gr.outputs.Textbox(label="Output length"),
187
+ # title="Training Interface",
188
+ # description="Enter some text and click the button to start training.",
189
+ # theme="default",
190
+ # layout="vertical",
191
+ # allow_flagging=False,
192
+ # allow_screenshot=False
193
+ # )
194
 
195
  if __name__ == '__main__':
196
+ # iface.launch(share=True)
197
+ train("Oi")
198
 
199
 
200