Update app.py

app.py

@@ -8,10 +8,6 @@ from basicsr.archs.srvgg_arch import SRVGGNetCompact
 from realesrgan.utils import RealESRGANer
 import base64 
 
-from dotenv import load_dotenv
-load_dotenv()
-
-
 model_realesr = SRVGGNetCompact(num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=32, upscale=4, act_type='prelu')
 model_path_realesr = 'realesr-general-x4v3.pth'
 
@@ -25,7 +21,7 @@ model_gfpgan_1_4 = GFPGANer(model_path='GFPGANv1.4.pth', upscale=2, arch='clean'
 
 
 os.makedirs('output', exist_ok=True)
-
+os.makedirs('temp', exist_ok=True)
 
 # def inference(img, version, scale, weight):
 def inference(img, version, scale):
@@ -90,8 +86,10 @@ def reconstruir_imagem():
         if token == "api_key_abcd":
 
             temp_filename = 'temp.jpg'
-            img_file.save(temp_filename)
+            pasta = f'temp/temp.jpg'
+            img_file.save(pasta)
 
+            # output, save_path = inference(temp_filename, version, scale)
             output, save_path = inference(temp_filename, version, scale)
 
             if output is not None:
@@ -112,5 +110,4 @@ def reconstruir_imagem():
         return jsonify({'status': 'error', 'message': str(e)})
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=80) 
-    
+    app.run(host='0.0.0.0', port=80)