manelhalima commited on
Commit
3412694
·
verified ·
1 Parent(s): 85b317d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -2,10 +2,14 @@ import gradio as gr
2
  from transformers import LlavaForConditionalGeneration, LlavaProcessor
3
  from PIL import Image
4
  import torch
 
 
5
 
6
  #Charger le modèle et le processeur
7
  model_id = "llava-hf/llava-1.5-7b-hf"
8
- processor = LlavaProcessor.from_pretrained(model_id)
 
 
9
  model = LlavaForConditionalGeneration.from_pretrained(
10
  model_id,
11
  torch_dtype=torch.float16,
 
2
  from transformers import LlavaForConditionalGeneration, LlavaProcessor
3
  from PIL import Image
4
  import torch
5
+ import warnings
6
+ warnings.filterwarnings("ignore")
7
 
8
  #Charger le modèle et le processeur
9
  model_id = "llava-hf/llava-1.5-7b-hf"
10
+ #processor = LlavaProcessor.from_pretrained(model_id)
11
+ processor = LlavaProcessor.from_pretrained(model_id, use_fast=False)
12
+
13
  model = LlavaForConditionalGeneration.from_pretrained(
14
  model_id,
15
  torch_dtype=torch.float16,