Adding model as input to gradio app

README.md

@@ -51,7 +51,7 @@ At minimum, this metric requires the model and references as inputs.
 
 Output Example(s):
 ```python
-{'accuracy': 1.0}
+{'loss': 4.993086338043213, 'perplexity': 147.390625}
 ```
 This metric outputs a dictionary, containing the loss and perplexity score.
 

dmx_perplexity/dmx_perplexity.py

@@ -51,6 +51,7 @@ class DmxPerplexity(evaluate.Metric):
             features=datasets.Features(
                 {
                     "references": datasets.Value("string"),
+                    "model":datasets.Value("string")
                 }
             ),
             reference_urls=["https://huggingface.co/docs/transformers/perplexity"],