test another model

language.py

@@ -1,9 +1,18 @@
 from transformers import pipeline
+import torch
 
 input = "i have question and answere.\
 the question is : How many cars here ?\
 the response is : 2\
 with this information, can you crate an answere phrase in french"
 
-generator = pipeline('text-generation', model='gpt2')
-print(generator(input, max_length=30))
+generate_text = pipeline(model="databricks/dolly-v2-12b", torch_dtype=torch.bfloat16, trust_remote_code=True, device_map="auto")
+res = generate_text(input)
+print(res[0]["generated_text"])
+
+
+
+#i have question and answere.
+#the question is : How many cars here ?
+#the response is : 2
+#with this information, can you crate an answere phrase

requirements.txt

@@ -1 +1,2 @@
-transformers[torch]
+transformers[torch]
+torch