Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,4 +5,6 @@ hf_token = os.environ.get("HF_TOKEN")
|
|
| 5 |
client = InferenceClient(model="bigcode/starcoderbase", token=hf_token)
|
| 6 |
|
| 7 |
def query_model(prompt):
|
| 8 |
-
return client.text_generation(prompt, max_new_tokens=100)
|
|
|
|
|
|
|
|
|
| 5 |
client = InferenceClient(model="bigcode/starcoderbase", token=hf_token)
|
| 6 |
|
| 7 |
def query_model(prompt):
|
| 8 |
+
return client.text_generation(prompt, max_new_tokens=100)
|
| 9 |
+
import os
|
| 10 |
+
print("Token Found:", bool(os.getenv("HF_TOKEN")))
|