cleanup

app.py

@@ -1,4 +1,3 @@
-import collections
 import datetime
 import os
 import random
@@ -9,7 +8,7 @@ import json
 
 import pandas as pd
 import ray
-from PIL import Image
+
 from compiled_jss.CPEnv import CompiledJssEnvCP
 
 from stable_baselines3.common.vec_env import VecEnvWrapper
@@ -21,8 +20,6 @@ import numpy as np
 from MyVecEnv import WrapperRay
 
 import gradio as gr
-import docplex.cp.utils_visu as visu
-import matplotlib.pyplot as plt
 
 
 class VecPyTorch(VecEnvWrapper):
@@ -155,17 +152,21 @@ ray.init(log_to_driver=False,
         ignore_reinit_error=True,
         include_dashboard=False)
 title = "Job-Shop Scheduling CP RL"
-description = "A Job-Shop Scheduling Reinforcement Learning based solver, using an underlying CP model as an " \
-              "environment. <br>" \
-              "However, due to resource limitations on the HuggingFace platform (a single vCPU available, no GPU), " \
-              "the results you obtain here don't represent the full potential of the approach. <br>" \
-              "For large instance, we recommend to run this locally outside the interface as it causes a lot of " \
-              "overhead. <br>" \
-              "For fast inference, check out the cached examples below. <br>" \
-              "Any Job-Shop Scheduling instance following the standard specification is compatible, <a href='http://jobshop.jjvh.nl/index.php'>checkout this website for more instances</a>." \
+description = """A Job-Shop Scheduling Reinforcement Learning based solver using an underlying CP model as an environment. <br>
+                However, the results you obtain here don't represent the full potential of the approach due to resource limitations on the HuggingFace platform (a single vCPU available, no GPU).<br>
+                We recommend running this locally outside the interface for large instances, as it causes a lot of overhead.<br>
+                For fast inference, check out the cached examples below.<br>
+                Any Job-Shop Scheduling instance following the standard specification is compatible. <a href='http://jobshop.jjvh.nl/index.php'>Check out this website for more instances</a>."""
 
 article = "<p style='text-align: center'>Article Under Review</p>"
 # list all non-hidden files in the 'instances' directory
 examples = ['instances/' + f for f in os.listdir('instances') if not f.startswith('.')]
-iface = gr.Interface(fn=solve, inputs=gr.File(label="Instance File"), outputs=[gr.Text(label="Makespan"), gr.Text(label="Elapsed Time"), gr.Text(label="Solution"), gr.Plot(label="Solution's Gantt Chart")], title=title, description=description, article=article, examples=examples)
+iface = gr.Interface(fn=solve,
+                     inputs=gr.File(label="Instance File"),
+                     outputs=[gr.Text(label="Makespan"), gr.Text(label="Elapsed Time"), gr.Text(label="Solution"), gr.Plot(label="Solution's Gantt Chart")],
+                     title=title,
+                     description=description,
+                     article=article,
+                     examples=examples,
+                     allow_flagging="never")
 iface.launch(enable_queue=True)