Tricia Nieva commited on
Commit
e4993f9
·
1 Parent(s): f9135e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -10,6 +10,9 @@ import gradio as gr
10
  from transformers import AutoModelForCausalLM, AutoTokenizer
11
  import torch
12
 
 
 
 
13
  openai.organization = "org-orRhfBkKOfOuNACbjPyWKbUt"
14
  openai.api_key = "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"
15
 
 
10
  from transformers import AutoModelForCausalLM, AutoTokenizer
11
  import torch
12
 
13
+ tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
14
+ model = AutoModelForCausalLM.from_pretrained("text-davinci-003")
15
+
16
  openai.organization = "org-orRhfBkKOfOuNACbjPyWKbUt"
17
  openai.api_key = "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"
18