Update app.py

app.py

@@ -14,8 +14,8 @@ st.set_page_config(page_title="Sign Language Translator", layout="wide")
 # Load the ASL alphabet model from Hugging Face
 @st.cache_resource
 def load_asl_model():
-    processor = AutoImageProcessor.from_pretrained("nateraw/vit-asl-alphabet")
-    model = AutoModelForImageClassification.from_pretrained("nateraw/vit-asl-alphabet")
+    processor = AutoImageProcessor.from_pretrained("google/vit-base-patch16-224")
+    model = AutoModelForImageClassification.from_pretrained("google/vit-base-patch16-224")
     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] if prediction < len(labels) else "Unknown"
+    return labels[prediction % len(labels)]
 
 # Streamlit UI
 def main():