Spaces:

AndreiFX
/

DreamDeep

Runtime error

AndreiFX commited on Jul 31, 2023

Commit

c328b26

1 Parent(s): 6b8cd3c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ prompt = os.environ["prompt"]
 client = Client(API_URL, headers={"Authorization": f"Bearer {hf_api_key}"}, timeout=120)
 def generate(input):
-    output = client.generate(prompt, max_new_tokens = 500).generated_text
     return output

 client = Client(API_URL, headers={"Authorization": f"Bearer {hf_api_key}"}, timeout=120)
 def generate(input):
+    output = client.generate(prompt + f'{input}', max_new_tokens = 500).generated_text
     return output