Update app.py

app.py

@@ -1,12 +1,8 @@
 
 import gradio as gr
-from transformers import TFAutoModelForSequenceClassification, AutoTokenizer
 import tensorflow as tf
 
-model_name = 'AiresPucrs/digit-classifier'
-
-model = TFAutoModelForSequenceClassification.from_pretrained(model_name)
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = "AiresPucrs/digit-classifier"
 
 classes = ['zero', 'one', 'two', 'three', 'four', 'five', 'six', 'seven', 'eight', 'nine']