code: def model_prediction(payload: CompletionPayload) -> QueueResponse:
      res = MODEL_API.add_to_queue(payload)
      return QueueResponse(qid=res["qid"])
  
file_path: mesh-transformer-jax/docker/main.py
imports: import uvicorn
  import traceback
  import logging
  from fastapi import FastAPI
  from starlette.middleware.cors import CORSMiddleware
  from .payloads import CompletionPayload, CompletionResponse, QueueRequest, QueueResponse
  from .ops import get_gptj_model
label_desc: Use image node_id 2 for calling a global function with example usage: model_prediction(payload) and returns: QueueResponse