appvoid commited on
Commit
9c665c5
Β·
verified Β·
1 Parent(s): 4c5c253

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,11 +7,11 @@ import torch
7
  from transformers import AutoModelForCausalLM, AutoTokenizer
8
  from transformers import pipeline
9
 
10
- model = pipeline("text-generation" , model="appvoid/palmer-004-2406")
11
 
12
  @spaces.GPU
13
  def predict(prompt):
14
- completion = model(prompt, max_new_tokens=64, temperature=0.2)[0]["generated_text"]
15
  return completion
16
 
17
  gr.Interface(
 
7
  from transformers import AutoModelForCausalLM, AutoTokenizer
8
  from transformers import pipeline
9
 
10
+ model = pipeline("text-generation" , model="appvoid/palmer-004")
11
 
12
  @spaces.GPU
13
  def predict(prompt):
14
+ completion = model(prompt, max_new_tokens=64, temperature=0.3)[0]["generated_text"]
15
  return completion
16
 
17
  gr.Interface(