Update app.py

app.py

@@ -4,12 +4,19 @@ import io
 import random
 import os
 from PIL import Image
+from deep_translator import GoogleTranslator
+from langdetect import detect
 
 API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
 API_TOKEN = os.getenv("HF_READ_TOKEN") # it is free
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
 
 def query(prompt, is_negative=False, steps=1, cfg_scale=6, seed=None):
+    language = detect(prompt)
+    
+    if language == 'ru':
+        prompt = GoogleTranslator(source='ru', target='en').translate(prompt)
+        print(prompt)
     payload = {
         "inputs": prompt,
         "is_negative": is_negative,