Update app.py

app.py

@@ -14,9 +14,9 @@ if st.button("Generate Image") and text:
     # Process text and get CLIP features
     text_features = clip_processor(text, return_tensors="pt", padding=True)
 
-    # Use CLIP's image projection to generate an image representation
-    image_representation = clip_model.get_image_features(pixel_values=text_features.pixel_values)
-    
+    # Use CLIP's encode_image method to obtain the image features
+    image_representation = clip_model.encode_image(text_features.pixel_values)
+
     # For visualization, you can convert the image representation back to an image
     image_array = image_representation.squeeze().permute(1, 2, 0).cpu().numpy()
     image = Image.fromarray((image_array * 255).astype('uint8'))