Update app.py

app.py

@@ -27,13 +27,8 @@ from safetensors.torch import load_file, save_file
 from diffusers import DiffusionPipeline, AnimateDiffPipeline, MotionAdapter, EulerDiscreteScheduler, DDIMScheduler, StableDiffusionXLPipeline, UNet2DConditionModel, AutoencoderKL, UNet3DConditionModel
 #import jax
 #import jax.numpy as jnp
-from numba import jit as cpu, cuda
-from numba.cuda import jit as gpu
-
-# optimization:
-
-# @gpu(cache=True)
-# @cpu(cache=True)
+from numba import cuda
+from numba.cuda import autojit as gpu, grid
 
 # logging
 
@@ -115,17 +110,10 @@ function custom(){
 
 # functionality
 
-@cpu(cache=True)
 def run(cmd):
-    try:
-        tx = cuda.threadIdx.x
-        bx = cuda.blockIdx.x
-        dx = cuda.blockDim.x
-        pos = tx + bx * dx
-    except:
-        pos = 0
-
-    cmd=cmd[pos]
+    #x = cuda.threadIdx.x + cuda.blockIdx.x * cuda.blockDim.x
+    #y = cuda.threadIdx.y + cuda.blockIdx.y * cuda.blockDim.y
+    #z = cuda.threadIdx.z + cuda.blockIdx.z * cuda.blockDim.z
 
     result = subprocess.run(cmd, shell=True, capture_output=True, env=None)
     if result.returncode != 0:
@@ -133,19 +121,14 @@ def run(cmd):
             f"Command '{cmd}' failed with exit status code '{result.returncode}'. Exiting..."
         )
         sys.exit()
-    return result
+    return str(result.stdout)
     
-@cpu(cache=True)
-def translate(args):
-    try:
-        tx = cuda.threadIdx.x
-        bx = cuda.blockIdx.x
-        dx = cuda.blockDim.x
-        pos = tx + bx * dx
-    except:
-        pos = 0
+@gpu()
+def translate(text,lang):
 
-    text=text[pos]
+    
+    
+    text=text[]
     lang=lang[pos]
 
     if text == None or lang == None: