Upload folder using huggingface_hub

README.md

@@ -1,12 +1,6 @@
 ---
-title: Diffuser2
-emoji: 👀
-colorFrom: yellow
-colorTo: red
+title: diffuser2
+app_file: main.py
 sdk: gradio
 sdk_version: 3.50.2
-app_file: app.py
-pinned: false
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

main.py

@@ -0,0 +1,15 @@
+from fastapi import FastAPI
+import gradio as gr
+import torch
+from diffusers import DiffusionPipeline
+import uvicorn
+
+description = "Image generation with GPT-2"
+title = "IMAGE GENERATION MACHINE"
+
+pipeline3 = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", use_safetensors=False).to('cuda')
+interface = gr.Interface(lambda x: pipeline3(x).images[0], 'textbox', 'image'
+#             description=description,
+#             examples=examples,
+)
+interface.launch(share = True)

requirements.txt

@@ -0,0 +1,6 @@
+fastapi
+torch==2.1.0
+uvicorn
+diffusers
+transformers
+gradio