Fix prediction function

app.py

@@ -21,7 +21,7 @@ def load_model(ckpt):
     return FlaxCLIPVisionBertForSequenceClassification.from_pretrained(ckpt)
 
 @st.cache(persist=True)
-def predict(model, transformed_image, question_inputs):
+def predict(transformed_image, question_inputs):
     return np.array(model(pixel_values = transformed_image, **question_inputs)[0][0])
     
 def softmax(logits):
@@ -125,7 +125,7 @@ state.answer_lang_id = col2.selectbox('Answer Language', index=options.index(sta
 with st.spinner('Loading model...'):
     model = load_model(checkpoints[0])
 with st.spinner('Predicting...'):
-    logits = predict(model, transformed_image, dict(question_inputs))
+    logits = predict(transformed_image, dict(question_inputs))
 logits = softmax(logits)
 labels, values = get_top_5_predictions(logits, answer_reverse_mapping)
 translated_labels = translate_labels(labels, state.answer_lang_id)