File size: 814 Bytes
5e1b2e8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
import gradio as gr
from PIL import Image
import os
import sys
# Add the project root directory to the Python path
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..')))
from src.anime_super_resolution.infer import infer
def run_inference(input_image, model_id, outer_scale):
if not input_image:
return None, "Please upload an image."
models_config_path = "configs/model_ckpts.yaml"
try:
output_image = infer(
input_path=input_image,
model_id=model_id,
models_config_path=models_config_path,
outer_scale=outer_scale,
)
return output_image, "Inference completed successfully!"
except Exception as e:
return None, f"Error during inference: {str(e)}" |