Update app.py

app.py

@@ -205,7 +205,7 @@ def process_best_match(best_match, language):
     best_story = best_match["Story"]
 
     # Translate to Arabic if the language is Arabic
-    if language == "Arabic":
+    if language == "Arabic" or language == "ar":
         best_story_translated = translate_story_to_arabic(best_story)
         info_html = f"<div dir='rtl' style='font-size: 18px; color: white; font-family: Arial, sans-serif;'>{best_story_translated}</div>"
         audio_file = text_to_speech_arabic(best_story_translated)
@@ -220,7 +220,8 @@ def process_best_match(best_match, language):
 def compare_images(image, language):
     try:
         inputs = processor(images=image, return_tensors="pt")
-        image_features = model.get_image_features(**inputs)
+        inputs = {k: v.to(device) for k, v in inputs.items()}
+        image_features = model.get_image_features(**inputs).to(device)
 
         best_score = -2.0
         best_match_idx = None