Reem1 commited on
Commit
fdf3c92
·
verified ·
1 Parent(s): f7fbebb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -5
app.py CHANGED
@@ -48,11 +48,6 @@ def generate_story(name: str, hobby: str):
48
  full_text = out[0]["generated_text"]
49
  story = full_text.split("[Assistant]:")[-1].replace("</s>", "").strip()
50
  return story
51
-
52
- out = pipe(prompt, max_new_tokens=900, temperature=0.9, top_p=0.95, do_sample=True)
53
- full_text = out[0]["generated_text"]
54
- story = full_text.split("[Assistant]:")[-1].replace("</s>", "").strip()
55
- return story
56
 
57
  iface = gr.Interface(
58
  fn=generate_story,
 
48
  full_text = out[0]["generated_text"]
49
  story = full_text.split("[Assistant]:")[-1].replace("</s>", "").strip()
50
  return story
 
 
 
 
 
51
 
52
  iface = gr.Interface(
53
  fn=generate_story,