davanstrien HF Staff commited on
Commit
f540eaf
·
1 Parent(s): 4102373

change default model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -170,7 +170,7 @@ with gr.Blocks() as demo:
170
  **Please note** the predictions will take some time since the model needs to be loaded for inference first. If you make a second batch of prediction using the same model the predictions should be quicker."""
171
  )
172
  token = gr.Textbox(label="token", type="password")
173
- model_id = gr.Textbox(label="model_id", value="microsoft/resnet-50")
174
  button = gr.Button("predict")
175
  gr.Markdown("## Results")
176
  plot = gr.BarPlot(x="labels", y="freqs", width=600, height=400, vertical=False)
 
170
  **Please note** the predictions will take some time since the model needs to be loaded for inference first. If you make a second batch of prediction using the same model the predictions should be quicker."""
171
  )
172
  token = gr.Textbox(label="token", type="password")
173
+ model_id = gr.Textbox(label="model_id", value="davanstrien/autotrain-wikiart-sample2-42615108993")
174
  button = gr.Button("predict")
175
  gr.Markdown("## Results")
176
  plot = gr.BarPlot(x="labels", y="freqs", width=600, height=400, vertical=False)