Lalitr commited on
Commit
df09b76
1 Parent(s): 280d8c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ def GetinfoLLAMA(query):
24
  response=requests.post(API_URL,headers=headers,json=payload)
25
  response=response.json()
26
  #return response[0]["generated_text"]
27
- return str(response[0]["generated_text"]).split('\n')[1]
28
 
29
  st.set_page_config(page_title='Joke Generator',
30
  page_icon=':circus_tent:',
 
24
  response=requests.post(API_URL,headers=headers,json=payload)
25
  response=response.json()
26
  #return response[0]["generated_text"]
27
+ return str(response[0]["generated_text"]).replace(query,'')
28
 
29
  st.set_page_config(page_title='Joke Generator',
30
  page_icon=':circus_tent:',