Update app.py
Browse files
app.py
CHANGED
|
@@ -3,6 +3,8 @@ import requests
|
|
| 3 |
import os # Import os to access environment variables
|
| 4 |
|
| 5 |
# Get API token from environment variable
|
|
|
|
|
|
|
| 6 |
|
| 7 |
# Change MODEL_ID to a better model
|
| 8 |
# MODEL_ID = "Salesforce/codet5p-770m" # CodeT5+ (Recommended)
|
|
|
|
| 3 |
import os # Import os to access environment variables
|
| 4 |
|
| 5 |
# Get API token from environment variable
|
| 6 |
+
API_TOKEN = os.getenv("HF_API_TOKEN")
|
| 7 |
+
|
| 8 |
|
| 9 |
# Change MODEL_ID to a better model
|
| 10 |
# MODEL_ID = "Salesforce/codet5p-770m" # CodeT5+ (Recommended)
|