Update app.py

app.py

@@ -4,9 +4,11 @@ from PIL import Image
 
 st.title('Take your picture to the next level')
 
+@st.cache_data
 def upscale(model_choice):
     if model_choice == "EDSR":
-        model = EdsrModel.from_pretrained('eugenesiow/edsr-base', scale=choice)
+        from transformers import AutoModel
+        model = AutoModel.from_pretrained("eugenesiow/edsr-base")
     else:
         st.text('Not supported yet. Comeback soon for fun releases')