jianjiang
/
vit-large-patch14-clip-224.openai-ft-in12k-in1k
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chang pre_train to VitProcess
Build-Deploy-Actions Details

This commit is contained in:
jianjiang 2023-05-04 11:10:35 +08:00
parent 9de5caeb3a
commit 2c44f1747a
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
app.py
View File

@ -1,5 +1,5 @@
import gradio as gr
from transformers import ViTImageProcessor, ViTModel
from transformers import ViTImageProcessor, ViTForImageClassification
from gradio.themes.utils import sizes
theme = gr.themes.Default(radius_size=sizes.radius_none).set(
block_label_text_color = '#4D63FF',
@ -11,7 +11,8 @@ theme = gr.themes.Default(radius_size=sizes.radius_none).set(
)
processor = ViTImageProcessor.from_pretrained('google/vit-large-patch16-224-in21k')
model = ViTModel.from_pretrained('google/vit-large-patch16-224-in21k')
#model = ViTModel.from_pretrained('google/vit-large-patch16-224-in21k')
model = ViTForImageClassification.from_pretrained('google/vit-large-patch16-224-in21k')
def image_classification(image):
inputs = processor(images=image, return_tensors="pt")
@ -31,4 +32,4 @@ demo = gr.Interface(fn=image_classification,
if __name__ == "__main__":
demo.queue(concurrency_count=10)
demo.launch(server_name = "0.0.0.0")
demo.launch(server_name = "0.0.0.0", share = True)