Trying dataset loading

app.py

@@ -1,13 +1,22 @@
 import cv2
 import numpy as np
 import streamlit as st
+from datasets import load_dataset
+from PIL import Image
+from matplotlib.pyplot import imshow
 
+def load_dataset_from_Hugging_Face():
+  reloaded_dataset_thermal = load_dataset("gokulraja17/rice-thermal-demo")
+  reloaded_dataset_rgb = load_dataset("gokulraja17/rice-rgb-demo2")
+  return reloaded_dataset_thermal,reloaded_dataset_rgb
+
+dataset_thermal,dataset_rgb = load_dataset_from_Hugging_Face()
 uploaded_file = st.file_uploader("Choose a image file", type=["jpg","png"])
 
 if uploaded_file is not None:
     # Convert the file to an opencv image.
     file_bytes = np.asarray(bytearray(uploaded_file.read()), dtype=np.uint8)
     opencv_image = cv2.imdecode(file_bytes, 1)
-
+    
     # Now do something with the image! For example, let's display it:
     st.image(opencv_image, channels="BGR")

requirements.txt

@@ -1 +1,4 @@
-opencv-python
+opencv-python
+transformers
+datasets
+matplotlib