Update app.py

app.py

@@ -6,9 +6,12 @@ from keras.models import Model
 import matplotlib.pyplot as plt
 import logging
 from skimage.transform import resize
-from PIL import Image, ImageEnhance
+from PIL import Image, ImageEnhance, ImageFilter
 from tqdm import tqdm
 
+# Ensure TensorFlow runs on CPU
+tf.config.set_visible_devices([], 'GPU')
+
 class SwarmAgent:
     def __init__(self, position, velocity):
         self.position = position
@@ -102,6 +105,12 @@ class SwarmNeuralNetwork:
         # Normalize to [0, 1] range for display
         self.generated_image = (self.generated_image + 1) / 2
         self.generated_image = np.clip(self.generated_image, 0, 1)
+        
+        # Apply sharpening filter
+        image_pil = Image.fromarray((self.generated_image * 255).astype(np.uint8))
+        image_pil = image_pil.filter(ImageFilter.SHARPEN)
+        image_pil = image_pil.filter(ImageFilter.DETAIL)
+        self.generated_image = np.array(image_pil) / 255.0
 
     def train(self, epochs):
         logging.basicConfig(filename='training.log', level=logging.INFO)