Hugging Face's logo

Spaces:
stardust23
/
ads-video-generator
Runtime error

App Files Community
jonathanjordan21 commited on
Commit
c55bf8f
·
1 Parent(s): 596977e

Change to smaller model

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -8,8 +8,11 @@ import os, gc
8
  import gradio as gr
9
 
10
 
11
- model = AutoModelForSeq2SeqLM.from_pretrained("declare-lab/flan-alpaca-gpt4-xl")
12
- tokenizer = AutoTokenizer.from_pretrained("declare-lab/flan-alpaca-gpt4-xl")
 
 
 
13
 
14
  pipe = pipeline(
15
  'text2text-generation',
 
8
  import gradio as gr
9
 
10
 
11
+ # model = AutoModelForSeq2SeqLM.from_pretrained("declare-lab/flan-alpaca-gpt4-xl")
12
+ # tokenizer = AutoTokenizer.from_pretrained("declare-lab/flan-alpaca-gpt4-xl")
13
+
14
+ model = AutoModelForSeq2SeqLM.from_pretrained("declare-lab/flan-alpaca-large")
15
+ tokenizer = AutoTokenizer.from_pretrained("declare-lab/flan-alpaca-large")
16
 
17
  pipe = pipeline(
18
  'text2text-generation',