User Hf model instead of vllm

app.py

@@ -1,13 +1,13 @@
 import gradio as gr
 import pandas as pd
 from typing import List, Dict
-from flow_judge import Vllm, FlowJudge, EvalInput
+from flow_judge import Hf, FlowJudge, EvalInput
 from flow_judge.metrics import CustomMetric, RubricItem
 
 try:
-    model = Vllm(quantized=False)
+    model = Hf()
 except Exception as e:
-    raise RuntimeError(f"Failed to initialize Vllm: {e}")
+    raise RuntimeError(f"Failed to initialize Hf Model: {e}")
 
 EXAMPLES = [
     {

requirements.txt

@@ -1,3 +1,2 @@
-vllm-flash-attn==2.6.2
-flow-judge[vllm]==0.1.0
+flow-judge[hf]==0.1.0
 flash_attn>=2.6.3