felfri commited on
Commit
d7d6a6e
1 Parent(s): b6bbd04

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -46,8 +46,8 @@ from transformers import AutoProcessor, LlavaForConditionalGeneration
46
  from PIL import Image
47
  import requests
48
 
49
- model = LlavaForConditionalGeneration.from_pretrained('LukasHug/LlavaGuard-13b-hf')
50
- processor = AutoProcessor.from_pretrained('LukasHug/LlavaGuard-13b-hf')
51
 
52
  conversation = [
53
  {
 
46
  from PIL import Image
47
  import requests
48
 
49
+ model = LlavaForConditionalGeneration.from_pretrained('AIML-TUDA/LlavaGuard-13B-hf')
50
+ processor = AutoProcessor.from_pretrained('AIML-TUDA/LlavaGuard-13B-hf')
51
 
52
  conversation = [
53
  {