Akbartus commited on
Commit
2304e1a
·
1 Parent(s): 902ae75

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ from transformers import AutoModel
12
  original_model = gr.Interface.load("models/Akbartus/Lora360")
13
 
14
  # Load the super-resolution model from Hugging Face's model hub
15
- super_resolution_model = AutoModel.from_pretrained('CompVis/ldm-super-resolution-4x-openimages')
16
 
17
  def process_image(input_image):
18
  # Run the original model
 
12
  original_model = gr.Interface.load("models/Akbartus/Lora360")
13
 
14
  # Load the super-resolution model from Hugging Face's model hub
15
+ super_resolution_model = AutoModel.from_pretrained('keras-io/super-resolution')
16
 
17
  def process_image(input_image):
18
  # Run the original model