Update app.py

app.py

@@ -1,22 +1,19 @@
 import gradio as gr
-from transformers import BlenderBotTokenizer, BlenderBotForConditionalGeneration
+from transformers import CLIPProcessor, CLIPModel
 
-# Load BlenderBot model and tokenizer
-model_name = "facebook/blenderbot-3B"
-tokenizer = BlenderBotTokenizer.from_pretrained(model_name)
-model = BlenderBotForConditionalGeneration.from_pretrained(model_name)
+# Load CLIP model and processor
+model_name = "openai/clip-vit-base-patch32"
+processor = CLIPProcessor.from_pretrained(model_name)
+model = CLIPModel.from_pretrained(model_name)
 
 def generate_3d_image(text_prompt):
     # Encode the text prompt
-    input_ids = tokenizer(text_prompt, return_tensors="pt").input_ids
+    inputs = processor(text=text_prompt, return_tensors="pt")
 
     # Generate image from text prompt
-    output = model.generate(input_ids)
+    output = model.get_image(inputs.pixel_values)
 
-    # Decode the generated image tokens
-    output_text = tokenizer.decode(output[0], skip_special_tokens=True)
-
-    return output_text
+    return "Image generated successfully based on the text prompt."
 
 # Create a Gradio interface for generating 3D images
 iface = gr.Interface(