Update app.py

app.py

@@ -6,7 +6,7 @@ from PIL import Image
 from replicate import Client
 
 
-illuse = replicate.Client(api_token=os.getenv('REPLICATE'))
+illuse = Client(api_token=os.getenv('REPLICATE'))
 model_name = "andreasjansson/illusion:75d51a73fce3c00de31ed9ab4358c73e8fc0f627dc8ce975818e653317cb919b"
 example_image = "https://replicate.delivery/pbxt/hHJNV9QteKX8DK2ckkUeXsqbEIKNGFXU1fN0MJoizz3iPlOjA/output-0.png"
 
@@ -36,8 +36,9 @@ def generate(prompt, negative_prompt, qr_content, pattern_image, num_inference_s
         return result
     except Exception as e:
         print(e)
-        gr.Error(str(e))
+        st.error(str(e))
         return
+        
 st.title("Illusion Diffusion Fast Demo powered by replicate")
 
 prompt = st.text_input("Prompt")