Update app.py

app.py

@@ -1,8 +1,9 @@
+# %%writefile demos/app.py
 ### 1. Imports and class names setup ### 
 import gradio as gr
 import os
 import torch
-from model import TinyCNN
+# from model import TinyCNN
 from timeit import default_timer as timer
 from typing import Tuple, Dict
 import torch
@@ -65,9 +66,9 @@ def predict(img) -> Tuple[Dict, float]:
 ### 4. Gradio app ###
 
 # Create title, description and article strings
-title = "Wafer Defect Detection"
-description = "An app to predict wafer Defects from the following 9 defects.[Center, Donut, Edge-Loc, Edge-Ring, Loc, Near-full, Random, Scratch, none]."
-
+title = "Expression Detection"
+description = "An app to predict emotions from the list.[Angry, Disgust, Fear, Happy, Neutral, Sad, Surprise]"
+article = "Created as a college  project."
 
 # Create examples list from "examples/" directory
 # example_list = [["examples/" + example] for example in os.listdir("examples")]
@@ -78,12 +79,13 @@ demo = gr.Interface(
     inputs=gr.Image(type="pil"),
     outputs=[
         gr.Label(num_top_classes=5, label="Predictions"),
-        gr.Image(label="Defects"),
+        gr.Number(label="Prediction time (s)"),
     ],
     # examples=example_list,
     title=title,
     description=description,
+    article=article,
 )
 
 # Launch the app!
-demo.launch(share=True)
+demo.launch()