Update app.py

app.py

@@ -1,20 +1,27 @@
-import gradio as gr
 from transformers import pipeline
+from flask import Flask, request, jsonify
 
+app = Flask(__name__)
 
-model_name = "joeddav/xlnet-base-cased"
-waifu2x = pipeline("image-upscaling", model=model_name)
+# Create the pipeline for image upscaling
+upscaler = pipeline("image-upscaling", model="deepset/electricity-converter-0036")
 
+# Define the route for the home page
+@app.route('/')
+def home():
+    return 'Image Upscaling App'
 
-def image_upscaler(input_image):
-    output_image = waifu2x(input_image)[0]['generated']
-    return output_image
+# Define the route for the image upscaling
+@app.route('/upscale', methods=['POST'])
+def upscale():
+    # Get the uploaded image
+    file = request.files['file']
 
+    # Upscale the image
+    output = upscaler(file)
 
-input_image = gr.inputs.Image(label="Input Image")
-output_image = gr.outputs.Image(label="Enlarged Image", type="numpy")
+    # Return the upscaled image
+    return jsonify({'image': output})
 
-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()
+if __name__ == '__main__':
+    app.run()