first commit

app.py

@@ -2,10 +2,9 @@ import streamlit as st
 from PIL import Image
 import kosmos
 
-kosmos.load_model()
-st.title('KOSMOS2 Model Evaluation')
+st.title('KOSMOS2 Chat box')
 
-uploaded_file = st.file_uploader("Choose an image...", type="jpg")
+uploaded_file = st.file_uploader("Pick an image...", type="jpg")
 
 if uploaded_file is not None:
     try:

kosmos.py

@@ -4,17 +4,10 @@ from os import path
 from torchvision.transforms import ToTensor
 
 model_id = "microsoft/kosmos-2-patch14-224"
-model = None
-processor = None
-
-def load_model():
-    global model, processor
-    model = AutoModelForVision2Seq.from_pretrained(model_id)
-    processor = AutoProcessor.from_pretrained(model_id)
+model = AutoModelForVision2Seq.from_pretrained(model_id)
+processor = AutoProcessor.from_pretrained(model_id)
 
 def single_image_classification(image, prompt="", max_new_tokens=30):
-    if model is None or processor is None:
-        load_model()
     inputs = processor(text=prompt, images=image, return_tensors="pt")
     
     generated_ids = model.generate(