Update app.py

app.py

@@ -3,7 +3,8 @@ import os
 from fastapi import FastAPI, Request
 from contextlib import asynccontextmanager
 from transformers import pipeline
-from langdetect import detect
+from langdetect import detect  # to change to langid
+import langid
 from huggingface_hub import login
 import socket
 
@@ -23,7 +24,8 @@ english_generator = pipeline("text-generation", model="distilgpt2")
 # Function to detect language
 def detect_language(user_input):
     try:
-        lang = detect(user_input)
+       # lang = detect(user_input)
+        lang = langid.classify(user_input)
         print(f"Detected language: {lang}") 
         return "hebrew" if lang == "he" else "english" if lang == "en" else "unsupported"
     except: