Update app.py

app.py

@@ -32,9 +32,19 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 model_id1 = "dreamlike-art/dreamlike-diffusion-1.0"
 pipe = StableDiffusionPipeline.from_pretrained(model_id1, torch_dtype=torch.float16, use_safetensors=True)
-pipe = pipe.to("cuda")
+pipe = pipe.to("device")
 
-…            temperature=0.7,
+# Updated function for text generation using the new API structure
+def generate_creative_text(prompt):
+    try:
+        response = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",  # Change this to the model you prefer, e.g., "gpt-4" if available
+            messages=[
+                {"role": "system", "content": "You are a creative assistant."},
+                {"role": "user", "content": prompt}
+            ],
+            max_tokens=1024,
+            temperature=0.7,
         )
         return response['choices'][0]['message']['content'].strip()
     except Exception as e:
@@ -76,7 +86,7 @@ def process_audio(audio_path, image_option, creative_text_option):
         try:
             model_id1 = "dreamlike-art/dreamlike-diffusion-1.0"
             pipe = StableDiffusionPipeline.from_pretrained(model_id1, torch_dtype=torch.float16, use_safetensors=True)
-            pipe = pipe.to("cuda")
+            pipe = pipe.to("device")
             image = pipe(translation).images[0]
         except Exception as e:
             return tamil_text, translation, creative_text, f"An error occurred during image generation: {str(e)}"