Update app.py

app.py

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