Update app.py

app.py

@@ -10,7 +10,7 @@ from PIL import Image
 import os
 
 token = os.environ["API_TOKEN"]
-
+model = InferenceClient("black-forest-labs/FLUX.1-schnell", token=token)
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -20,15 +20,20 @@ MAX_IMAGE_SIZE = 2048
 def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, progress=gr.Progress(track_tqdm=True)):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
-    model =  InferenceClient("black-forest-labs/FLUX.1-schnell", token=prompt)
-    image = model(prompt)
+    image = client.text_to_image(
+        "A girl and a boy dancing in the forest",
+        height=height,
+        width=width,
+        seed=seed,
+        num_inference_steps=4
+    )
     temp_dir = tempfile.gettempdir()
     temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".jpg", dir=temp_dir)
     image.save(temp_file, format="JPEG")
     temp_file_path = temp_file.name
     temp_file.close()
     return temp_file_path, seed
- 
+
 examples = [
     "a tiny astronaut hatching from an egg on the moon",
     "a cat holding a sign that says hello world",