Update app.py

app.py

@@ -232,14 +232,12 @@ def handle_query():
 ### استعلام SQL (SELECT فقط):
 """
 
+        # تحضير المدخلات بدون تحويل نوع البيانات مباشرة
         inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=512)
         
-        # تحويل المدخلات إلى float32 إذا كنا على CPU
-        if not torch.cuda.is_available():
-            inputs = inputs.to(torch.float32)
-        
-        if torch.cuda.is_available():
-            inputs = inputs.to('cuda')
+        # نقل المدخلات إلى الجهاز المناسب (CPU/GPU)
+        device = 'cuda' if torch.cuda.is_available() else 'cpu'
+        inputs = {k: v.to(device) for k, v in inputs.items()}
         
         with torch.no_grad():
             outputs = model.generate(
@@ -250,6 +248,8 @@ def handle_query():
             )
             
         sql = tokenizer.decode(outputs[0], skip_special_tokens=True)
+        logger.info(f"⚡ الاستعلام المولد: {sql}")
+        
         sql = clean_sql(sql)
         
         if not is_safe_sql(sql):