jonACE commited on
Commit
710e3c3
·
verified ·
1 Parent(s): e7c9e58

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -105,11 +105,11 @@ tokenizer.push_to_hub("jonACE/llama-2-7b-chat_fine_tuned", token=hf_token)
105
 
106
 
107
  # CHATBOT START
108
- chatbot = pipeline("text-generation", model="jonACE/llama-2-7b-chat_fine_tuned")
109
 
110
- def chatbot_response(prompt):
111
- result = chatbot(prompt, max_length=100, do_sample=True, temperature=0.7)
112
- return result[0]["generated_text"]
113
 
114
- iface = gr.Interface(fn=chatbot_response, inputs="text", outputs="text")
115
- iface.launch()
 
105
 
106
 
107
  # CHATBOT START
108
+ # chatbot = pipeline("text-generation", model="jonACE/llama-2-7b-chat_fine_tuned")
109
 
110
+ #def chatbot_response(prompt):
111
+ # result = chatbot(prompt, max_length=100, do_sample=True, temperature=0.7)
112
+ # return result[0]["generated_text"]
113
 
114
+ #iface = gr.Interface(fn=chatbot_response, inputs="text", outputs="text")
115
+ #iface.launch()