Update app.py

app.py

@@ -272,6 +272,16 @@ def infer(ref_style_file, style_description, caption):
         # Reset the state after inference, regardless of success or failure
         reset_inference_state()
 
+def transform(image):
+    """
+    Define the necessary transformations for the image.
+    """
+    return T.Compose([
+        T.Resize((256, 256)),  # Example resize, adjust as needed
+        T.ToTensor(),
+        T.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
+    ])(image)
+
 def infer_compo(style_description, ref_style_file, caption, ref_sub_file):
     global models_rbm, models_b
     try: