Update output gradio

app.py

@@ -39,7 +39,7 @@ def text_predictor(title, synopsis):
     for prob, cls in zip(genres, class_names):
         if prob >= 0.5:
             preds_name.append(cls)
-    return round(score.item(), 2), isAward.item(), {"genres":preds_name}
+    return {'score':round(score.item(), 2), 'award_winning':isAward.item(), "genres":preds_name}
     
 def img_predictor(img):
     # Preprocess the image
@@ -56,7 +56,7 @@ def img_predictor(img):
         if prob >= 0.5:
             preds_name.append(cls)
 
-    return round(score.item(), 2), isAward.item(), {"genres": preds_name}
+    return {'score':round(score.item(), 2), 'award_winning':isAward.item(), "genres":preds_name}
 
 
 def combine_predictor(title, synopsis, img):
@@ -83,14 +83,14 @@ def combine_predictor(title, synopsis, img):
         if prob >= 0.5:
             preds_name.append(cls)
             
-    return round(score.item(), 2), isAward.item(), {"genres": preds_name}
+    return {'score':round(score.item(), 2), 'award_winning':isAward.item(), "genres":preds_name}
     
 # iface_1 = gr.Interface(age_predictor_image, gr.Image(height=256, width=256), "json", examples=[["young.webp"], ["old.jpg"]])
-iface_1 = gr.Interface(text_predictor, [gr.Text(placeholder="Input title here"), gr.Text(placeholder="Input synopsis here")], ["label", "label", "json"])
+iface_1 = gr.Interface(text_predictor, [gr.Text(placeholder="Input title here"), gr.Text(placeholder="Input synopsis here")], ["json"])
 
-iface_2 = gr.Interface(img_predictor, gr.Image(height=224, width=224), ["label", "label", "json"])
+iface_2 = gr.Interface(img_predictor, gr.Image(height=224, width=224), ["json"])
 
-iface_3 = gr.Interface(combine_predictor, [gr.Text(placeholder="Input title here"), gr.Text(placeholder="Input synopsis here"), gr.Image(height=224, width=224)], ["label", "label", "json"])
+iface_3 = gr.Interface(combine_predictor, [gr.Text(placeholder="Input title here"), gr.Text(placeholder="Input synopsis here"), gr.Image(height=224, width=224)], ["json"])
 demo = gr.TabbedInterface([iface_1, iface_2, iface_3], ["From Text", "From Image", "From Text and Image"])
 demo.launch()  # Launches the mini app!
     

dockerfile

@@ -1,30 +0,0 @@
-# read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
-# you will also find guides on how best to write your Dockerfile
-
-FROM python:3.9
-
-WORKDIR /code
-
-COPY ./requirements.txt ./requirements.txt
-
-COPY ./model_only.pt ./model_only.pt
-
-RUN pip install --no-cache-dir --upgrade -r ./requirements.txt
-
-# Set up a new user named "user" with user ID 1000
-RUN useradd -m -u 1000 user
-
-# Switch to the "user" user
-USER user
-
-# Set home to the user's home directory
-ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:$PATH
-
-# Set the working directory to the user's home directory
-WORKDIR $HOME/app
-
-# Copy the current directory contents into the container at $HOME/app setting the owner to the user
-COPY --chown=user . $HOME/app
-
-CMD ["python", "app.py"]