first

app.py

@@ -3,7 +3,6 @@ from ulti import *
 import threading
 import pipeline
 import streamlit as st
-import numpy as np
 from PIL import Image
 
 
@@ -24,8 +23,8 @@ def main():
         if img_file_buffer is not None:
             from PIL import Image
             col1, col2 = st.columns(2)
-            image = Image.open(img_file_buffer)
-            image = np.array(image)
+            image = Image.open(img_file_buffer).convert("RGB")
+            #image = np.array(image)
             # image = adjust_gamma(image, gamma=gamma)
             with col1:
                 st.image(image)

pipeline.py

@@ -24,8 +24,8 @@ poem = pipeline('text-generation', model="./GPT2/rkw_4sen",
                 config=config)
 
 
-def main(img_path):
-    clses = vis_extractor.predict(img_path)
+def main(img):
+    clses = vis_extractor.predict(img)
     
     keywords = clses
     print(keywords)

query2labels/infer.py

@@ -216,8 +216,8 @@ class Query2Label():
         return 
 
     @torch.no_grad()
-    def predict(self, image_path):
-        image = Image.open(image_path).convert("RGB")
+    def predict(self, image):
+        #image = Image.open(image_path).convert("RGB")
         test_data_transform = transforms.Compose([
             transforms.Resize((self.args.img_size, self.args.img_size)),
             transforms.ToTensor()])

requirements.txt

@@ -5,5 +5,4 @@ yapf
 timm
 transformers
 termcolor
-scipy
-numpy
+scipy