Update app.py

app.py

@@ -1,12 +1,11 @@
 from pdf2image import convert_from_path
 import pandas as pd, numpy as np
-from PIL import Image
-import pickle
+#from PIL import Image
+#import pickle
 import gradio as gr
 from transformers import pipeline
 import pandas as pd
-from sentence_transformers import SentenceTransformer, util
-import pickle
+#from sentence_transformers import SentenceTransformer, util
 import pytesseract
 from transformers import pipeline
 pipe = pipeline("text-classification",model="acrowth/autotrain-preesmetextclassifier-2437575785")
@@ -16,6 +15,11 @@ pipe = pipeline("text-classification",model="acrowth/autotrain-preesmetextclassi
 #    pipe=pickle.load(file)
 
 def findpdftype(file):
+    images = convert_from_path(file.name, first_page=1,last_page=1)
+    images[0].save('test.jpg')
+    return pipe('test.jpg')[0]['label']
+
+def findpdftypeold2(file):
     images = convert_from_path(file.name)
     results=[]
     for image in images: