Nlpeva commited on
Commit
218fe85
·
verified ·
1 Parent(s): a9f950d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -2,7 +2,14 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
2
  from peft import PeftModel
3
  import torch
4
  import gradio as gr
5
- from huggingface_hub import spaces
 
 
 
 
 
 
 
6
 
7
  # --- Model Loading ---
8
  base_model_id = "unsloth/Meta-Llama-3.1-8B"
 
2
  from peft import PeftModel
3
  import torch
4
  import gradio as gr
5
+ from huggingface_hub import login, spaces
6
+
7
+ # --- Authentication ---
8
+ try:
9
+ login()
10
+ print("Logged in to Hugging Face Hub successfully!")
11
+ except Exception as e:
12
+ print(f"Error during Hugging Face login: {e}")
13
 
14
  # --- Model Loading ---
15
  base_model_id = "unsloth/Meta-Llama-3.1-8B"