TanmayTomar commited on
Commit
4b1fd1f
·
verified ·
1 Parent(s): a0a6620

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -43,7 +43,7 @@ AutoModelForSequenceClassification.from_pretrained('MoritzLaurer/DeBERTa-v3-base
43
  from transformers import T5Tokenizer, T5ForConditionalGeneration; \
44
  T5ForConditionalGeneration.from_pretrained('google/flan-t5-base', cache_dir='/app/cache'); \
45
  from transformers import pipeline; \
46
- pipeline('image-classification', model='umm-maybe/AI-image-detector', cache_dir='/app/cache'); \
47
  "
48
 
49
  # Copy the rest of your application code
 
43
  from transformers import T5Tokenizer, T5ForConditionalGeneration; \
44
  T5ForConditionalGeneration.from_pretrained('google/flan-t5-base', cache_dir='/app/cache'); \
45
  from transformers import pipeline; \
46
+ pipeline('image-classification', model='umm-maybe/AI-image-detector'); \
47
  "
48
 
49
  # Copy the rest of your application code