edit app.py

app.py

@@ -13,7 +13,7 @@ tfg_FONT_demo = gr.Interface(fn=tfg_FONT, inputs=[gr.Image(type="filepath"), gr.
                               ["./example/images/7.png", "./example/audios/6350921755403330389_00056.wav"]])
 
 def tfg_OPT(source_img, driving_audio):
-    tfg = OPT()
+    tfg = OPT(checkpoint_path='./Demo_TFR_Pirenderer/checkpoints', config_path='./Demo_TFR_Pirenderer/src/config')
     video_path = tfg.test(source_image=source_img, driven_audio=driving_audio)
     return video_path