Spaces:
Sleeping
Sleeping
SHAMIL SHAHBAZ AWAN
commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -32,7 +32,7 @@ if not hf_api_key:
|
|
| 32 |
else:
|
| 33 |
# Initialize the model and tokenizer using the API key
|
| 34 |
try:
|
| 35 |
-
model_name = "
|
| 36 |
model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=hf_api_key)
|
| 37 |
tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=hf_api_key)
|
| 38 |
llm_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer)
|
|
|
|
| 32 |
else:
|
| 33 |
# Initialize the model and tokenizer using the API key
|
| 34 |
try:
|
| 35 |
+
model_name = "llama-3.1-8b-instant" # Replace with the correct model name if needed
|
| 36 |
model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=hf_api_key)
|
| 37 |
tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=hf_api_key)
|
| 38 |
llm_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer)
|