fix app

app.py

@@ -3,6 +3,7 @@ import cv2
 import gradio as gr
 import random
 import numpy as np
+import pandas as pd
 tflite_filename = 'model-400.tflite'
 interpreter = tf.lite.Interpreter(model_path=tflite_filename)
 runner = interpreter.get_signature_runner()
@@ -12,9 +13,8 @@ init_states = {
 }
 del init_states['image']
 
-with open('labels.txt', 'r') as f:
-    CLASSES = f.read().splitlines()
-
+data = pd.read_csv('labels.csv', header=None)
+CLASSES = data[1].values
 def format_frames(frame, output_size):
     """
         Pad and resize an image from a video.