Update app.py

app.py

@@ -1,16 +1,20 @@
 import gradio as gr
 from transformers import pipeline
 
-model_name = "waifu2x-multi-lang"
+
+model_name = "joeddav/xlnet-base-cased"
 waifu2x = pipeline("image-upscaling", model=model_name)
 
 
-def waifu2x_enlarge(input_image):
-    output_image = waifu2x(input_image)
+def image_upscaler(input_image):
+    output_image = waifu2x(input_image)[0]['generated']
     return output_image
 
 
 input_image = gr.inputs.Image(label="Input Image")
 output_image = gr.outputs.Image(label="Enlarged Image", type="numpy")
 
-gr.Interface(waifu2x_enlarge, input_image, output_image).launch()
+title = "Waifu2x Image Upscaler"
+description = "This app uses the Hugging Face transformers library to upscale images using the Waifu2x model."
+
+gr.Interface(fn=image_upscaler, inputs=input_image, outputs=output_image, title=title, description=description).launch()