Update app.py

app.py

@@ -828,12 +828,6 @@
 # demo.launch(share=True)
 
 
-
-
-
-
-
-
 import gradio as gr
 import requests
 import os
@@ -1035,28 +1029,12 @@ def generate_audio_parler_tts(text, chunk_size=200):
     logging.debug(f"Audio saved to {temp_audio_path}")
     return temp_audio_path
 
+# Initialize the Stable Diffusion pipeline
+pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4").to("cuda")
 
-def transcribe_function(stream, new_chunk):
-    try:
-        sr, y = new_chunk[0], new_chunk[1]
-    except TypeError:
-        print(f"Error chunk structure: {type(new_chunk)}, content: {new_chunk}")
-        return stream, "", None
-
-    y = y.astype(np.float32) / np.max(np.abs(y))
-
-    if stream is not None:
-        stream = np.concatenate([stream, y])
-    else:
-        stream = y
-
-    result = pipe_asr({"array": stream, "sampling_rate": sr}, return_timestamps=False)
-
-    full_text = result.get("text", "")
-    
-    return stream, full_text, result
-
-
+def generate_image(prompt):
+    image = pipe(prompt).images[0]
+    return image
 
 with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
 
@@ -1097,3 +1075,11 @@ demo.queue()
 demo.launch(share=True)
 
 
+
+
+
+
+
+
+
+