Update app.py

app.py

@@ -19,7 +19,10 @@ pipe = StableDiffusionPipeline.from_pretrained(
 
 pipe = pipe.to(device)
 
-def text_to_image(prompt, save_as):
+def text_to_image(prompt, save_as, key_id):
+
+    if AWS_ACCESS_KEY_ID != key_id:
+        return "not permition"
 
     # Create an instance of the S3 client
     s3 = boto3.client('s3',
@@ -52,5 +55,5 @@ def text_to_image(prompt, save_as):
 
 
 
-iface = gr.Interface(fn=text_to_image, inputs=[Textbox(label="prompt"), Textbox(label="s3_save_as")], outputs="text")
+iface = gr.Interface(fn=text_to_image, inputs=[Textbox(label="prompt"), Textbox(label="s3_save_as"), Textbox(label="aws_key_id")], outputs="text")
 iface.launch()