app.py updates

app.py

@@ -1,3 +1,4 @@
+from datasets.arrow_dataset import InMemoryTable
 import streamlit as st
 from PIL import Image, ImageDraw
 
@@ -45,11 +46,11 @@ def patch(i):
   img=i//(36*36)
   p=i%(36*36)
 
-  im=ds[img]['image']
+  imm=ds[img]['image']
 
   x=p//36
   y=p%36
-  imc=im.crop((x*16,y*16,(x+1)*16,(y+1)*16))
+  imc=imm.crop((x*16,y*16,(x+1)*16,(y+1)*16))
   return imc