Update app.py

app.py

@@ -24,6 +24,10 @@ import jax.numpy as jnp
 import sys
 import warnings
 
+class Suppress:
+    def write(self, msg):
+        pass
+
 warnings.filterwarnings("ignore")
 sys.stderr = Suppress()
 
@@ -86,14 +90,6 @@ function custom(){
 }
 """
 
-#def forest_schnell():
-#    PIPE = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, token=os.getenv("hf_token")).to("cuda")
-#    return PIPE
-
-class DevNull:
-    def write(self, msg):
-        pass
-
 def translate(text,lang):
     if text == None or lang == None:
         return ""