Harshithtd commited on
Commit
acc7d19
1 Parent(s): 9b4901f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -7,10 +7,11 @@ from gradio import FileData
7
  import time
8
  import spaces
9
  # Load model directly
 
10
  from transformers import AutoProcessor, AutoModelForPreTraining
11
 
12
- processor = AutoProcessor.from_pretrained("meta-llama/Llama-3.2-11B-Vision-Instruct")
13
- model = AutoModelForPreTraining.from_pretrained("meta-llama/Llama-3.2-11B-Vision-Instruct")
14
 
15
  @spaces.CPU
16
  def bot_streaming(message, history, max_new_tokens=250):
 
7
  import time
8
  import spaces
9
  # Load model directly
10
+ # Load model directly
11
  from transformers import AutoProcessor, AutoModelForPreTraining
12
 
13
+ processor = AutoProcessor.from_pretrained("unsloth/Llama-3.2-11B-Vision-Instruct-bnb-4bit")
14
+ model = AutoModelForPreTraining.from_pretrained("unsloth/Llama-3.2-11B-Vision-Instruct-bnb-4bit")
15
 
16
  @spaces.CPU
17
  def bot_streaming(message, history, max_new_tokens=250):