Update app.py

app.py

@@ -1,9 +1,8 @@
-# %%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
@@ -66,12 +65,12 @@ def predict(img) -> Tuple[Dict, float]:
 ### 4. Gradio app ###
 
 # Create title, description and article strings
-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."
+title = "Wafer Defect Detection"
+description = "An app to predict Wafer Defects in semiconductors.[Center, Donut, Edge-Loc, Edge-Ring, Loc, Near-full, Random, Scratch, none]"
+
 
 # Create examples list from "examples/" directory
-# example_list = [["examples/" + example] for example in os.listdir("examples")]
+example_list = [["example/" + example] for example in os.listdir("example")]
 
 # Create Gradio interface 
 demo = gr.Interface(
@@ -81,10 +80,10 @@ demo = gr.Interface(
         gr.Label(num_top_classes=5, label="Predictions"),
         gr.Number(label="Prediction time (s)"),
     ],
-    # examples=example_list,
+    examples=example_list,
     title=title,
     description=description,
-    article=article,
+
 )
 
 # Launch the app!