zineb-chgari commited on
Commit
5d66efe
·
verified ·
1 Parent(s): 950b91e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -14,6 +14,7 @@ from transformers import pipeline
14
  # (Keep Constants as is)
15
  # --- Constants ---
16
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
17
  print("Token loaded:", bool(os.getenv("chatbotagenthf")))
18
  custom_prompt = """
19
  You are an intelligent AI agent participating in the GALA (Generative Agent Learning Assignment).
 
14
  # (Keep Constants as is)
15
  # --- Constants ---
16
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
17
+ pipe = pipeline("text-generation", model="WeiboAI/VibeThinker-1.5B")
18
  print("Token loaded:", bool(os.getenv("chatbotagenthf")))
19
  custom_prompt = """
20
  You are an intelligent AI agent participating in the GALA (Generative Agent Learning Assignment).