Spaces:

gustavoaq
/

llama_7b_ft_2

Paused

gustavoaq commited on Apr 9, 2023

Commit

adfcdb4

1 Parent(s): d445807

Update finetune.py

Files changed (1) hide show

finetune.py CHANGED Viewed

@@ -171,7 +171,7 @@ model.state_dict = (
 import gradio as gr
-def train():
     print(os.listdir(OUTPUT_DIR))
     # Call your trainer's train() function here
     trainer.train()
@@ -193,7 +193,7 @@ iface = gr.Interface(
 )
 if __name__ == '__main__':
-    iface.launch()

 import gradio as gr
+def train(input_text):
     print(os.listdir(OUTPUT_DIR))
     # Call your trainer's train() function here
     trainer.train()
 )
 if __name__ == '__main__':
+    iface.launch(share=True)