Spaces:

cody82
/

bot_innopolis

Running

cody82 commited on Jul 21

Commit

9116b46

verified ·

1 Parent(s): 7cd7e31

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import torch
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-model_id = "google/flan-t5-base"  # можно flan-t5-large, если хватает памяти
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_id)
 device = "cuda" if torch.cuda.is_available() else "cpu"

 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+model_id = "openai-community/gpt2"  # можно flan-t5-large, если хватает памяти
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_id)
 device = "cuda" if torch.cuda.is_available() else "cpu"