Update app.py
Browse files
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()
|