far555na commited on
Commit
b78b8e7
·
verified ·
1 Parent(s): 00dd512

Update ig.py

Browse files
Files changed (1) hide show
  1. ig.py +5 -2
ig.py CHANGED
@@ -1,10 +1,13 @@
1
  from transformers import AutoProcessor, AutoModelForCausalLM
2
  import gradio as gr
3
  import torch
 
4
 
5
- # Load the processor and model
 
 
6
  processor = AutoProcessor.from_pretrained("microsoft/git-base")
7
- model = AutoModelForCausalLM.from_pretrained("./")
8
 
9
  def predict(image):
10
  try:
 
1
  from transformers import AutoProcessor, AutoModelForCausalLM
2
  import gradio as gr
3
  import torch
4
+ from transformers import AutoAdapterModel, AutoTokenizer
5
 
6
+ model_name = "./"
7
+ model = AutoAdapterModel.from_pretrained(model_name)
8
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
9
  processor = AutoProcessor.from_pretrained("microsoft/git-base")
10
+
11
 
12
  def predict(image):
13
  try: