Abubakar Abid commited on
Commit
8c1a379
1 Parent(s): 4b24692

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -6,5 +6,6 @@ os.system("pip install gradio==2.5.8a0")
6
  import gradio as gr
7
 
8
  HF_TOKEN = os.getenv('HF_TOKEN')
9
- hf_saver = gr.HuggingFaceDatasetSaver(HF_TOKEN, "test-image-classifier-dataset")
10
- gr.Interface.load("huggingface/google/vit-base-patch16-224", flagging_callback=hf_saver).launch()
 
 
6
  import gradio as gr
7
 
8
  HF_TOKEN = os.getenv('HF_TOKEN')
9
+ description = "Translate something from English to French. Flag to add to this dataset: https://huggingface.co/datasets/abidlabs/test-image-classifier-dataset."
10
+ hf_saver = gr.HuggingFaceDatasetSaver(HF_TOKEN, "test-translation-dataset")
11
+ gr.Interface.load("huggingface/t5-base", flagging_callback=hf_saver).launch()