Update app.py

app.py

@@ -11,11 +11,11 @@ from transformers import AutoImageProcessor, AutoModelForImageClassification
 # Set page config as the first command
 st.set_page_config(page_title="Sign Language Translator", layout="wide")
 
-# Load the open-source ASL model from Hugging Face
+# Load the ASL alphabet model from Hugging Face
 @st.cache_resource
 def load_asl_model():
-    processor = AutoImageProcessor.from_pretrained("google/vit-base-patch16-224-in21k")
-    model = AutoModelForImageClassification.from_pretrained("google/vit-base-patch16-224-in21k")
+    processor = AutoImageProcessor.from_pretrained("Roboflow/ASL-Alphabet-Classifier")
+    model = AutoModelForImageClassification.from_pretrained("Roboflow/ASL-Alphabet-Classifier")
     return processor, model
 
 processor, model = load_asl_model()
@@ -27,7 +27,7 @@ def classify_asl(image):
     outputs = model(**inputs)
     prediction = torch.argmax(outputs.logits, dim=-1).item()
     labels = list("ABCDEFGHIJKLMNOPQRSTUVWXYZ")  # ASL alphabet labels
-    return labels[prediction % len(labels)]
+    return labels[prediction] if prediction < len(labels) else "Unknown"
 
 # Streamlit UI
 def main():