morenolq commited on
Commit
e6612b7
1 Parent(s): 1624dcd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -2,10 +2,9 @@ import gradio as gr
2
  from transformers import pipeline
3
  from transformers import AutoTokenizer, AutoModelForCausalLM
4
 
5
- tokenizer = AutoTokenizer.from_pretrained("facebook/galactica-1.3b")
6
- model = AutoModelForCausalLM.from_pretrained("facebook/galactica-1.3b")
7
-
8
- text2text_generator = pipeline("text-generation", model=model, tokenizer=tokenizer, num_workers=2)
9
 
10
  def predict(text):
11
  text = text.strip()
 
2
  from transformers import pipeline
3
  from transformers import AutoTokenizer, AutoModelForCausalLM
4
 
5
+ #tokenizer = AutoTokenizer.from_pretrained("facebook/galactica-1.3b")
6
+ #model = AutoModelForCausalLM.from_pretrained("facebook/galactica-1.3b")
7
+ #text2text_generator = pipeline("text-generation", model=model, tokenizer=tokenizer, num_workers=2)
 
8
 
9
  def predict(text):
10
  text = text.strip()