Update app.py

app.py

@@ -8,6 +8,7 @@ from diffusers import DiffusionPipeline
 import torch
 import spaces  # Hugging Face Spaces module
 
+import easyocr
 import requests
 
 
@@ -45,7 +46,11 @@ def generate_caption_and_image(image):
     # reader = easyocr.Reader(['en'])  
     # result = reader.readtext(img)
     import random
-    
+    reader = easyocr.Reader(['ur', 'eng'])
+
+    # Read text from an image
+    result = reader.readtext(img)
+        
 
     # Define lists for the three variables
     fabrics = ['cotton', 'silk', 'denim', 'linen', 'polyester', 'wool', 'velvet']
@@ -71,7 +76,7 @@ def generate_caption_and_image(image):
     out = model.generate(**inputs)
     caption1 = processor.decode(out[0], skip_special_tokens=True)
 
-    prompt = f'''Create a highly realistic clothing item based on the following descriptions: The design should reflect {caption1} and {caption2}, blending both themes into a single, stylish, and modern piece of clothing. Incorporate highly realistic and high-quality textures that exude sophistication, with realistic fabric lighting and fine details. Subtly hint at {selected_fabric}, featuring a {selected_pattern} motif and a {selected_textile_design} style that harmoniously balances the essence of both captions.'''
+    prompt = f'''Create a highly realistic clothing item based on the following descriptions: The design should reflect {caption1} and {caption2}, blending both themes into a single, stylish, and modern piece of clothing. Incorporate highly realistic and high-quality textures that exude sophistication, with realistic fabric lighting and fine details. Subtly hint at {selected_fabric}, featuring a {selected_pattern} motif and a {selected_textile_design} style that harmoniously balances the essence of both captions.and {result} should be written on top of it'''