tutorials
/
resnet-dogcat
forked from ailab/resnet-dogcat
1
0
Fork 0
Code Pull Requests Activity

更新 README.md

This commit is contained in:
ailab 2024-05-27 14:21:12 +08:00
parent 30ca939316
commit 8726b97d8e
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
README.md
View File

@ -54,6 +54,7 @@ print(model.config.id2label[predicted_label])
``` ```
```python ```python
from transformers import pipeline
import gradio as gr import gradio as gr
pipe = pipeline("image-classification", model="ailab/resnet-dogcat") pipe = pipeline("image-classification", model="ailab/resnet-dogcat")
gr.Interface.from_pipeline(pipe).launch(server_name="0.0.0.0") gr.Interface.from_pipeline(pipe).launch(server_name="0.0.0.0")