BeastGokul commited on
Commit
e329667
·
verified ·
1 Parent(s): fee1e40

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,11 +5,11 @@ from peft import PeftModel, PeftConfig
5
  import os
6
  client = InferenceClient(api_key=(os.getenv('HF_TOKEN')))
7
 
8
- tokenizer = AutoTokenizer.from_pretrained("BeastGokul/Bio-Mistral-7B-finetuned")
9
  base_model = AutoModelForCausalLM.from_pretrained("BeastGokul/Bio-Mistral-7B-finetuned")
10
  base_model.resize_token_embeddings(len(tokenizer))
11
  model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.3")
12
-
13
  import gradio as gr
14
  from huggingface_hub import InferenceClient
15
 
 
5
  import os
6
  client = InferenceClient(api_key=(os.getenv('HF_TOKEN')))
7
 
8
+
9
  base_model = AutoModelForCausalLM.from_pretrained("BeastGokul/Bio-Mistral-7B-finetuned")
10
  base_model.resize_token_embeddings(len(tokenizer))
11
  model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.3")
12
+ tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.3")
13
  import gradio as gr
14
  from huggingface_hub import InferenceClient
15