add 3 inputs for function and rename from greet to model_call

app.py

@@ -24,7 +24,7 @@ Your output should be in JSON FORMAT with the keys "REASONING" and "SCORE":
 """
 
 
-def greet(question, document, answer):
+def model_call(question, document, answer):
     NEW_FORMAT = PROMPT.format(question=question, document=document, answer=answer)
     tokenizer = AutoTokenizer.from_pretrained("PatronusAI/Llama-3-Patronus-Lynx-8B-Instruct")
     model = AutoModelForCausalLM.from_pretrained("PatronusAI/Llama-3-Patronus-Lynx-8B-Instruct", cache_dir='/tmp/cache', torch_dtype=torch.float16, low_cpu_mem_usage=True)
@@ -34,5 +34,11 @@ def greet(question, document, answer):
     print(generated_text)
     return generated_text
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
+inputs = [
+    gr.Textbox(label="Question"),
+    gr.Textbox(label="Document"),
+    gr.Textbox(label="Answer")
+]
+
+demo = gr.Interface(fn=model_call, inputs=inputs, outputs="text")
 demo.launch()