Update app.py

app.py

@@ -2,8 +2,9 @@ import streamlit as st
 import tensorflow as tf
 from tensorflow_addons.layers import InstanceNormalization
 import numpy as np
-from PIL import Image
+import matplotlib.pyplot as plt
 from huggingface_hub import hf_hub_download
+from PIL import Image
 import requests
 from io import BytesIO
 import os
@@ -63,15 +64,11 @@ repo_name = "CycleGAN-Model"
 repo_id = f"{username}/{repo_name}"
 model_filename = "CycleGAN.h5"
 
-try:
-    # Download the model file from the repository
-    model_path = hf_hub_download(repo_id=repo_id, filename=model_filename)
-    # Load the model
-    model = tf.keras.models.load_model(model_path, custom_objects={"InstanceNormalization": InstanceNormalization})
-    st.success("Model loaded successfully!")
-except Exception as e:
-    st.error(f"An error occurred while loading the model: {e}")
-    st.stop()
+# Download the model file from the repository
+model_path = hf_hub_download(repo_id=repo_id, filename=model_filename)
+
+# Load the model
+model = tf.keras.models.load_model(model_path, custom_objects={"InstanceNormalization": InstanceNormalization})
 
 def load_and_preprocess_image(image_path_or_url):
     if isinstance(image_path_or_url, str) and image_path_or_url.startswith(('http://', 'https://')):