Sebastien De Greef commited on
Commit
c0f4fad
1 Parent(s): 457af58

Add auth button

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. requirements.txt +1 -0
app.py CHANGED
@@ -149,6 +149,7 @@ def save_model(model_name, hub_model_name, hub_token, gguf_16bit, gguf_8bit, ggu
149
  with gr.Blocks(title="Unsloth fine-tuning") as demo:
150
  with gr.Column():
151
  gr.Image("unsloth.png", width="300px", interactive=False, show_download_button=False, show_label=False)
 
152
  with gr.Column():
153
  gr.Markdown(f"**User:** {hf_user}\n\n**GPU Information:** {gpu_stats.name} ({max_memory} GB)\n\n[Unsloth Docs](http://docs.unsloth.com/)\n\n[Unsloth GitHub](https://github.com/unslothai/unsloth)")
154
  with gr.Tab("Base Model Parameters"):
 
149
  with gr.Blocks(title="Unsloth fine-tuning") as demo:
150
  with gr.Column():
151
  gr.Image("unsloth.png", width="300px", interactive=False, show_download_button=False, show_label=False)
152
+ gr.LoginButton()
153
  with gr.Column():
154
  gr.Markdown(f"**User:** {hf_user}\n\n**GPU Information:** {gpu_stats.name} ({max_memory} GB)\n\n[Unsloth Docs](http://docs.unsloth.com/)\n\n[Unsloth GitHub](https://github.com/unslothai/unsloth)")
155
  with gr.Tab("Base Model Parameters"):
requirements.txt CHANGED
@@ -5,4 +5,5 @@ peft
5
  accelerate
6
  bitsandbytes
7
  gradio
 
8
  tensorboard
 
5
  accelerate
6
  bitsandbytes
7
  gradio
8
+ gradio[oauth]
9
  tensorboard