Update app.py

app.py

@@ -6,6 +6,8 @@ import cv2
 import numpy as np
 import tensorflow.compat.v1 as tf
 import os
+import streamlit as st
+
 
 model = torch.hub.load('/home/ai/Downloads/yolov5', 'custom', path='/home/ai/Desktop/Icevision/yoloocr_best.engine', source='local')
 model.cuda()