Update app.py

app.py

@@ -32,7 +32,7 @@ pipe.load_lora_weights(custom_model, use_auth_token=True)
 #pipe.to("cuda")
 pipe.enable_model_cpu_offload()
 
-def infer(image_in):
+def infer(image_in, prompt):
     prompt = "aerial view, a futuristic research complex in a bright foggy jungle, hard lighting"
     negative_prompt = 'low quality, bad quality, sketches'