= commited on
Commit
2f9e713
1 Parent(s): ebdd158

fixed a goof

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -1,15 +1,12 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- # generator = pipeline(model="cactusfriend/nightmare-invokeai-prompts",
5
- # tokenizer="cactusfriend/nightmare-invokeai-prompts",
6
- # task="text-generation")
7
-
8
- generator = pipeline(model="/Users/ryan/python/dunston/v25/final",
9
- tokenizer="/Users/ryan/python/dunston/v25/final",
10
  task="text-generation")
11
 
12
 
 
13
  def makePrompts(prompt: str, temp: float, p: float, k: int):
14
  temp = float(temp)
15
  outputs = generator(prompt, max_new_tokens=200,
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ generator = pipeline(model="cactusfriend/nightmare-invokeai-prompts",
5
+ tokenizer="cactusfriend/nightmare-invokeai-prompts",
 
 
 
 
6
  task="text-generation")
7
 
8
 
9
+
10
  def makePrompts(prompt: str, temp: float, p: float, k: int):
11
  temp = float(temp)
12
  outputs = generator(prompt, max_new_tokens=200,