testing gradio

app.py

@@ -1,3 +1,27 @@
 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()

requirements.txt

@@ -355,7 +355,6 @@ tokenize-rt==5.2.0
 tomlkit==0.12.0
 toolz==0.12.0
 torch==2.1.0
-torchrl @ git+https://github.com/pytorch/rl.git@bf264e0e24971fc05ec42b571de7b8df84043a51
 torchsnapshot==0.1.0
 torchvision==0.16.0
 tornado==6.3.3