Spaces:
Runtime error
Runtime error
=
commited on
Commit
·
2f9e713
1
Parent(s):
ebdd158
fixed a goof
Browse files
app.py
CHANGED
@@ -1,15 +1,12 @@
|
|
1 |
import gradio as gr
|
2 |
from transformers import pipeline
|
3 |
|
4 |
-
|
5 |
-
|
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,
|