yonkoyonks commited on
Commit
b835094
·
verified ·
1 Parent(s): 8ebabee

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -46,7 +46,7 @@ Answer (with explanation):
46
 
47
  # Use Hugging Face InferenceClient directly
48
  client = InferenceClient(token=os.environ.get("HUGGINGFACE_API_KEY"))
49
- response = client.text_generation(model="google/gemma-2b-it", inputs=prompt)
50
 
51
  # Extract the generated text from the response
52
  answer = response[0]["generated_text"] if isinstance(response, list) and "generated_text" in response[0] else str(response)
 
46
 
47
  # Use Hugging Face InferenceClient directly
48
  client = InferenceClient(token=os.environ.get("HUGGINGFACE_API_KEY"))
49
+ response = client.text_generation(prompt, model="google/gemma-2b-it")
50
 
51
  # Extract the generated text from the response
52
  answer = response[0]["generated_text"] if isinstance(response, list) and "generated_text" in response[0] else str(response)