s0uL141 commited on
Commit
98abd16
·
verified ·
1 Parent(s): a9177d7
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ load_dotenv()
9
  groq_api_key = os.environ.get("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")
10
 
11
  # Initialize the Hugging Face model
12
- huggingface_model_name = "s0uL141/Cyber_gemma2_2B_it" # Your Hugging Face model name
13
  tokenizer = AutoTokenizer.from_pretrained(huggingface_model_name)
14
  huggingface_model = AutoModelForCausalLM.from_pretrained(huggingface_model_name)
15
 
 
9
  groq_api_key = os.environ.get("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")
10
 
11
  # Initialize the Hugging Face model
12
+ huggingface_model_name = "s0uL141/fine_tuned_science_gemma2b-it" # Your Hugging Face model name
13
  tokenizer = AutoTokenizer.from_pretrained(huggingface_model_name)
14
  huggingface_model = AutoModelForCausalLM.from_pretrained(huggingface_model_name)
15