taesiri commited on
Commit
32cf6d9
1 Parent(s): 7fe7653
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -19,8 +19,8 @@ import torchvision
19
  from huggingface_hub import HfApi, login, snapshot_download
20
  from PIL import Image
21
 
22
- # session_token = os.environ.get("SessionToken")
23
- # login(token=session_token)
24
 
25
  csv.field_size_limit(sys.maxsize)
26
 
@@ -365,7 +365,7 @@ with gr.Blocks(css=newcss, theme=gr.themes.Soft()) as demo:
365
  label_plot = gr.Plot(
366
  label="Is this a correct label for this image?", type="fig"
367
  )
368
- with gr.Column(scale=3):
369
  training_samples = gr.Gallery(
370
  type="pil", label="Training samples", elem_id="sample_gallery"
371
  )
 
19
  from huggingface_hub import HfApi, login, snapshot_download
20
  from PIL import Image
21
 
22
+ session_token = os.environ.get("SessionToken")
23
+ login(token=session_token)
24
 
25
  csv.field_size_limit(sys.maxsize)
26
 
 
365
  label_plot = gr.Plot(
366
  label="Is this a correct label for this image?", type="fig"
367
  )
368
+ with gr.Column(scale=2):
369
  training_samples = gr.Gallery(
370
  type="pil", label="Training samples", elem_id="sample_gallery"
371
  )