app.py

import gradio as gr
from train import TrainingLoop



def image_classifier(inp):
    return {'cat': 0.3, 'dog': 0.7}





iface = gr.Interface(fn=image_classifier, inputs="image", outputs="label")




load_trained = gr.Interface(fn=image_classifier, inputs="image", outputs="label")




attribute = gr.Interface(fn=image_classifier, inputs="image", outputs="label")


# iface = gr.TabbedInterface(interface_list=[train_from_scratch, load_trained, attribute], tab_names=["Train from Scratch", "Continue Training", "Attribute"],title="Attribution in Deep Reinforcement Learning")
iface.launch()