Spaces:
Sleeping
Sleeping
Update utils.py
Browse files
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"
|
| 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)
|