Liori25 commited on
Commit
a0ab790
·
verified ·
1 Parent(s): d5694d9

Update IO_pipeline.py

Browse files
Files changed (1) hide show
  1. IO_pipeline.py +2 -7
IO_pipeline.py CHANGED
@@ -12,15 +12,10 @@ from huggingface_hub import InferenceClient
12
  class RecipeDigitalizerPipeline:
13
  def __init__(self):
14
  print("Connecting to Hugging Face API (Qwen Mode)...")
15
- # Ensure HF_TOKEN is set in your environment
16
  self.token = os.getenv("HF_TOKEN")
17
-
18
- if not self.token:
19
- print("⚠️ WARNING: HF_TOKEN is missing. The API call will likely fail.")
20
 
21
- # --- WE ARE STICKING TO QWEN ---
22
- # If 2.5 gives you trouble, you can try "Qwen/Qwen2-VL-7B-Instruct"
23
- self.model_id = "Qwen/Qwen2.5-VL-7B-Instruct"
24
 
25
  self.client = InferenceClient(token=self.token)
26
 
 
12
  class RecipeDigitalizerPipeline:
13
  def __init__(self):
14
  print("Connecting to Hugging Face API (Qwen Mode)...")
 
15
  self.token = os.getenv("HF_TOKEN")
 
 
 
16
 
17
+ # This targets the larger model which uses a DIFFERENT server
18
+ self.model_id = "Qwen/Qwen2.5-VL-72B-Instruct"
 
19
 
20
  self.client = InferenceClient(token=self.token)
21