Set environment variables for CUDA and Hugging Face paths; increase image prompt height and enable sharing for demo launch

app.py

@@ -9,6 +9,7 @@ import os
 os.environ['SPCONV_ALGO'] = 'native'
 from typing import *
 import torch
+print(torch.__version__)
 import numpy as np
 import imageio
 import uuid
@@ -239,7 +240,7 @@ with gr.Blocks(title="Game Items Generator") as demo:
                 item_text_field = gr.Textbox(label="Item Name", placeholder="Enter the name of the item", lines=2, scale=4)
                 enhance_prompt_btn = gr.Button("Enhance Prompt", variant="primary", scale=1)
             generate_image_btn = gr.Button("Generate Image", variant="primary")
-            image_prompt = gr.Image(label="Image Prompt", image_mode="RGBA", type="pil", height=300)
+            image_prompt = gr.Image(label="Image Prompt", image_mode="RGBA", type="pil", height=400)
             
             with gr.Accordion(label="Generation Settings", open=False):
                 seed = gr.Slider(0, MAX_SEED, label="Seed", value=0, step=1)
@@ -358,4 +359,4 @@ if __name__ == "__main__":
         pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8)))    # Preload rembg
     except:
         pass
-    demo.launch()
+    demo.launch(share=True)