zerogpu

app.py

@@ -1,12 +1,9 @@
 import gradio as gr
 import pandas as pd
 from green_score import GREEN
+import spaces
 
-# Include the GREEN class code here
-# -------------------------------------
-# Paste the GREEN class code above this line
-# -------------------------------------
-
+@spaces.GPU  # Add the GPU decorator for functions that need GPU access
 def run_green(ref_text, hyp_text, model_name="StanfordAIMI/GREEN-radllama2-7b"):
     refs = [ref_text.strip()]
     hyps = [hyp_text.strip()]
@@ -54,7 +51,7 @@ with gr.Blocks(title="GREEN Score Evaluation Demo") as demo:
         choice = gr.Radio(
             label="Choose Input Type",
             choices=["Custom"] + list(examples.keys()),
-            value="Example 1",
+            value="Custom",
             interactive=True
         )
 

requirements.txt

@@ -1 +1,2 @@
-green_score
+green_score
+spaces