upload v1664846447 model

README.md

@@ -1,8 +1,7 @@
 ---
 title: "test"
-description: "test"
 sdk: "gradio"
 app_file: app.py
 tags: ["tfx", "auto-generated"]
 pinned: false
----
+---

app.py

@@ -8,9 +8,9 @@ from huggingface_hub import from_pretrained_keras
 PRETRAIN_CHECKPOINT = "google/vit-base-patch16-224-in21k"
 feature_extractor = ViTFeatureExtractor.from_pretrained(PRETRAIN_CHECKPOINT)
 
-# $chansung/vit-e2e-pipeline-hf-integration should be like chansung/test-vit
-# $v1664841204 
-MODEL_CKPT = "chansung/vit-e2e-pipeline-hf-integration@v1664841204"
+# test_model should be like chansung/test-vit
+# test_version 
+MODEL_CKPT = "test_model@test_version"
 MODEL = from_pretrained_keras(MODEL_CKPT)
 
 RESOLTUION = 224
@@ -51,12 +51,8 @@ def get_predictions(image: Image) -> tf.Tensor:
 
     confidences = {labels[i]: float(probs[0][i]) for i in range(3)}
     return confidences
-    
-title = "Simple demo for a Image Classification of the Beans Dataset with HF ViT model"
-
-description = """
 
-"""
+title = "Simple demo for a Image Classification of the Beans Dataset with HF ViT model"
 
 demo = gr.Interface(
     get_predictions,
@@ -66,4 +62,4 @@ demo = gr.Interface(
     title=title,
 )
 
-demo.launch(debug=True)    
+demo.launch(debug=True)