adert

Author: Dhanush-R-git

.gitignore

@@ -160,5 +160,6 @@ cython_debug/
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
-
-.env/
+myenv/
+.env/
+Dockerfile

app.py

@@ -36,7 +36,13 @@
 )
 
 repo_id = "mistralai/Mistral-7B-Instruct-v0.3"
-llm = HuggingFaceEndpoint(repo_id=repo_id, max_length=1024, temperature=0.3, token=os.getenv('HUGGINGFACEHUB_API_TOKEN'))
+llm = HuggingFaceEndpoint(
+    repo_id=repo_id,
+    task="summarization",
+    max_length=1024, 
+    temperature=0.3, 
+    token=os.getenv('HUGGINGFACEHUB_API_TOKEN')
+    )
 
 # Define the prompt template
 prompt_template = """