Spaces:
Runtime error
Runtime error
from flask import Flask, request, jsonify, render_template | |
from PIL import Image | |
import requests | |
from io import BytesIO | |
import base64 | |
app = Flask(__name__) | |
def home(): | |
return render_template("index.html") | |
def submit(): | |
cloth = request.files['cloth'] | |
model = request.files['model'] | |
## replace the url from the ngrok url provided on the notebook on server. | |
url = "https://5a37436f4b54b9c2cd.gradio.live" | |
print("sending") | |
response = requests.post(url=url, files={"cloth":cloth.stream, "model":model.stream}) | |
op = Image.open(BytesIO(response.content)) | |
buffer = BytesIO() | |
op.save(buffer, 'png') | |
buffer.seek(0) | |
data = buffer.read() | |
data = base64.b64encode(data).decode() | |
return render_template('index.html', op=data) | |
# return render_template('index.html', test=True) | |
if __name__ == '__main__': | |
app.run(debug=True) | |