Upload model_process.py

model_process.py

@@ -47,6 +47,7 @@ def get_result(question, data, example=None):
     os.makedirs(f"temp/{dir_path}", exist_ok=True)
     raw_answer = sess.get("https://yundisk.de/d/OneDrive_5G/Pic/data.json").json()
     if question in raw_answer:
+        raw_answer=raw_answer[question]
         classes = raw_answer["classes"]
         text_inputs = torch.cat([tokenizer(f"a photo of {c}") for c in classes])
         if raw_answer["need_example"]: