commit

README.md

@@ -1,10 +1,10 @@
 ---
-title: Aiclown
-emoji: 🐠
+title: Demo Illusion Diffusion Hq
+emoji: 📉
 colorFrom: indigo
 colorTo: indigo
 sdk: gradio
-sdk_version: 3.50.2
+sdk_version: 3.44.4
 app_file: app.py
 pinned: false
 ---

app.py

@@ -0,0 +1,77 @@
+import gradio as gr
+import boto3
+import banana_dev as client
+import os
+from typing import Iterable
+import gradio as gr
+from gradio.themes.base import Base
+from gradio.themes.utils import colors, fonts, sizes
+from PIL import Image
+import io
+import base64
+from io import BytesIO
+import traceback
+import uuid
+from dotenv import load_dotenv
+import gzip
+
+load_dotenv()
+
+BANANA_API_KEY = os.getenv("BANANA_API_KEY")
+BANANA_MODEL_KEY = os.getenv("BANANA_MODEL_KEY")
+BANANA_URL = os.getenv("BANANA_URL")
+
+def predict(image, prompt,qr_code, negative_prompt, seed):
+    try:
+        my_model = client.Client(
+            api_key=BANANA_API_KEY,
+            model_key=BANANA_MODEL_KEY,
+            url=BANANA_URL,
+        )
+        #target image
+        with open(image.name, "rb") as f:
+            image_bytes = f.read()
+        image_encoded = base64.b64encode(image_bytes)
+        image = image_encoded.decode("utf-8")
+
+        print(image)
+        print(negative_prompt)
+        print(seed)
+        print(qr_code)
+        print(prompt)
+
+        inputs = {
+            "prompt" : prompt,
+            "qr_code_content" : qr_code,
+            "negative_prompt": negative_prompt,
+            "image": image,
+            "seed": seed
+        }
+        result, meta = my_model.call("/", inputs)
+        output_bytes = result["outputs"]
+
+        image_encoded = output_bytes.encode('utf-8')
+        image_data = base64.b64decode(image_encoded)
+        image_data = gzip.decompress(image_data)
+        image_io = BytesIO(image_data)
+        output_image = Image.open(image_io)
+        return output_image, None
+    except Exception as e:
+        tb_str = traceback.format_exception(type(e), e, e.__traceback__)
+        tb_str = "".join(tb_str)
+        return None, tb_str
+
+with gr.Blocks() as demo:                         
+    title_with_logo = gr.Markdown("# Powered by Banana <img src='https://files.umso.co/lib_udQIfMXlGlflDxFd/rnqmq7j8oqug1kae.png' align='left' width='50' height='50'>")
+    with gr.Column():                        
+        image = gr.File(label="Image", file_count="single", scale=1)
+        prompt=gr.Textbox(label="Prompt")
+        qr_code=gr.Textbox(label="QR code link")
+        negative_prompt= gr.Textbox(label="Negative Prompt")
+        seed= gr.Textbox(label="Seed")
+        btn = gr.Button("Run illusion-diffusion-hq")                       
+        output_image = gr.Image(label="Output Image")  
+        output_error = gr.Textbox(label="Error Message")  
+    btn.click(predict, [image, prompt,qr_code, negative_prompt, seed], [output_image, output_error])
+
+demo.launch()

requirements.txt

@@ -0,0 +1,62 @@
+aiofiles==23.2.1
+altair==5.1.1
+annotated-types==0.5.0
+anyio==3.7.1
+attrs==23.1.0
+banana-dev==5.1.0
+boto3==1.28.50
+botocore==1.31.50
+certifi==2023.7.22
+charset-normalizer==3.2.0
+click==8.1.7
+contourpy==1.1.1
+cycler==0.11.0
+fastapi==0.103.1
+ffmpy==0.3.1
+filelock==3.12.4
+fonttools==4.42.1
+fsspec==2023.9.1
+gradio==3.44.4
+gradio_client==0.5.1
+h11==0.14.0
+httpcore==0.18.0
+httpx==0.25.0
+huggingface-hub==0.17.2
+idna==3.4
+importlib-resources==6.0.1
+Jinja2==3.1.2
+jmespath==1.0.1
+jsonschema==4.19.0
+jsonschema-specifications==2023.7.1
+kiwisolver==1.4.5
+MarkupSafe==2.1.3
+matplotlib==3.8.0
+numpy==1.26.0
+orjson==3.9.7
+packaging==23.1
+pandas==2.1.0
+Pillow==10.0.1
+pydantic==2.3.0
+pydantic_core==2.6.3
+pydub==0.25.1
+pyparsing==3.1.1
+python-dateutil==2.8.2
+python-multipart==0.0.6
+pytz==2023.3.post1
+PyYAML==6.0.1
+referencing==0.30.2
+requests==2.31.0
+rpds-py==0.10.3
+s3transfer==0.6.2
+semantic-version==2.10.0
+six==1.16.0
+sniffio==1.3.0
+starlette==0.27.0
+toolz==0.12.0
+tqdm==4.66.1
+typing_extensions==4.8.0
+tzdata==2023.3
+urllib3==1.26.16
+uvicorn==0.23.2
+websockets==11.0.3
+python-dotenv