Update app.py

app.py

@@ -25,7 +25,7 @@ class Predictor:
             )
         os.chdir('..')
 
-        """Load the model into memory to make running multiple predictions efficient"""
+        """💎 Load the model into memory to make running multiple predictions efficient"""
         self.face_swapper = insightface.model_zoo.get_model('models/inswapper_128.onnx',
                                                             providers=onnxruntime.get_available_providers())
         self.face_enhancer = gfpgan.GFPGANer(model_path='models/GFPGANv1.4.pth', upscale=1)
@@ -38,11 +38,11 @@ class Predictor:
             largest = max(analysed, key=lambda x: (x.bbox[2] - x.bbox[0]) * (x.bbox[3] - x.bbox[1]))
             return largest
         except:
-            print("No face found")
+            print("❌ No face found")
             return None
 
     def predict(self, input_image_path, swap_image_path):
-        """Run a single prediction on the model"""
+        """🚫 Run a single prediction on the model"""
         try:
             frame = cv2.imread(input_image_path)
             face = self.get_face(frame)
@@ -50,7 +50,7 @@ class Predictor:
             try:
                 print(frame.shape, face.shape, source_face.shape)
             except:
-                print("printing shapes failed.")
+                print("💊 printing shapes failed.")
             result = self.face_swapper.get(frame, face, source_face, paste_back=True)
 
             _, _, result = self.face_enhancer.enhance(
@@ -67,7 +67,7 @@ class Predictor:
 
 # Instantiate the Predictor class
 predictor = Predictor()
-title = "Swap Faces Using Our Model!!!"
+title = "# 🧸 Auto adjust Models."
 
 # Create Gradio Interface
 iface = gr.Interface(