Update app.py
Browse files
app.py
CHANGED
@@ -161,7 +161,6 @@ def run_inference(process_id, input_image, upscale_factor, seed, num_inference_s
|
|
161 |
logger.info("Inference completed for process_id: %s", process_id)
|
162 |
|
163 |
@app.route('/infer', methods=['POST'])
|
164 |
-
@spaces.GPU
|
165 |
def infer():
|
166 |
# Check if the file was provided in the form-data
|
167 |
if 'input_image' not in request.files:
|
@@ -218,7 +217,6 @@ def infer():
|
|
218 |
|
219 |
# Modify status endpoint to receive process_id in request body
|
220 |
@app.route('/status', methods=['POST'])
|
221 |
-
@spaces.GPU
|
222 |
def status():
|
223 |
data = request.json
|
224 |
process_id = data.get('process_id')
|
@@ -254,6 +252,6 @@ def status():
|
|
254 |
})
|
255 |
|
256 |
if __name__ == '__main__':
|
257 |
-
app.run(debug=True)
|
258 |
|
259 |
|
|
|
161 |
logger.info("Inference completed for process_id: %s", process_id)
|
162 |
|
163 |
@app.route('/infer', methods=['POST'])
|
|
|
164 |
def infer():
|
165 |
# Check if the file was provided in the form-data
|
166 |
if 'input_image' not in request.files:
|
|
|
217 |
|
218 |
# Modify status endpoint to receive process_id in request body
|
219 |
@app.route('/status', methods=['POST'])
|
|
|
220 |
def status():
|
221 |
data = request.json
|
222 |
process_id = data.get('process_id')
|
|
|
252 |
})
|
253 |
|
254 |
if __name__ == '__main__':
|
255 |
+
app.run(debug=True,host='0.0.0.0')
|
256 |
|
257 |
|