JuliaUpton commited on
Commit
b1b2671
1 Parent(s): a669ecc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,11 +1,11 @@
1
  import gradio as gr
2
  from transformers import AutoTokenizer
3
- from peft import PeftModel, AutoAdapterModel
4
 
5
  # Load the tokenizer and model
6
  model_name = "JuliaUpton/Math_AI"
7
  mixtral_tokenizer = AutoTokenizer.from_pretrained(model_name)
8
- merged_model = AutoAdapterModel.from_pretrained(model_name)
9
 
10
  # Define the input formatting function
11
  def input_from_text(instruction):
 
1
  import gradio as gr
2
  from transformers import AutoTokenizer
3
+ from peft import PeftModel
4
 
5
  # Load the tokenizer and model
6
  model_name = "JuliaUpton/Math_AI"
7
  mixtral_tokenizer = AutoTokenizer.from_pretrained(model_name)
8
+ merged_model = PeftModel.from_pretrained(model_name)
9
 
10
  # Define the input formatting function
11
  def input_from_text(instruction):